diff --git a/src/main/kotlin/org/domaindrivenarchitecture/provs/desktop/infrastructure/Firefox.kt b/src/main/kotlin/org/domaindrivenarchitecture/provs/desktop/infrastructure/Firefox.kt
index e2c2a57..8d1130d 100644
--- a/src/main/kotlin/org/domaindrivenarchitecture/provs/desktop/infrastructure/Firefox.kt
+++ b/src/main/kotlin/org/domaindrivenarchitecture/provs/desktop/infrastructure/Firefox.kt
@@ -38,4 +38,5 @@ fun Prov.installFirefox() = task {
     )
 
     aptInstall("firefox")
+    cmd("apt upgrade -y firefox", sudo = true)
 }
diff --git a/src/test/kotlin/org/domaindrivenarchitecture/provs/desktop/infrastructure/FirefoxKtTest.kt b/src/test/kotlin/org/domaindrivenarchitecture/provs/desktop/infrastructure/FirefoxKtTest.kt
index 3cf7acf..4381a6b 100644
--- a/src/test/kotlin/org/domaindrivenarchitecture/provs/desktop/infrastructure/FirefoxKtTest.kt
+++ b/src/test/kotlin/org/domaindrivenarchitecture/provs/desktop/infrastructure/FirefoxKtTest.kt
@@ -9,13 +9,15 @@ internal class FirefoxKtTest {
 
     @ExtensiveContainerTest
     fun installFirefox() {
-        // when
-        val res = defaultTestContainer().installFirefox()
+        defaultTestContainer().session {
+            // when
+            val res = installFirefox()
 
-        // then
-        assertTrue(res.success)
+            // then
+            assertTrue(res.success)
 
-        val ffIsInstalled = defaultTestContainer().isPackageInstalled("firefox")
-        assertTrue(ffIsInstalled)
+            val ffIsInstalled = isPackageInstalled("firefox")
+            assertTrue(ffIsInstalled)
+        }
     }
 }