diff --git a/test/integration/main_test.rb b/test/integration/main_test.rb
index f878ed08a..5cc061692 100644
--- a/test/integration/main_test.rb
+++ b/test/integration/main_test.rb
@@ -2,17 +2,11 @@
 
 class MainTest < IntegrationTest
   test "deploy, redeploy, rollback, details and audit" do
-    first_version = latest_app_version
+    first_version = update_app_rev
 
     assert_app_is_down
 
-    mock = Minitest::Mock.new
-    mock.expect(:manifest, false, [ first_version ])
-    mock.expect(:installed?, true)
-    mock.expect(:running?, true)
-    Kamal::Commands::Docker.stub(:new, mock) do
-      kamal :deploy
-    end
+    kamal :deploy
     assert_app_is_up version: first_version
     assert_hooks_ran "pre-connect", "pre-build", "pre-deploy", "post-deploy"
     assert_envs version: first_version
@@ -21,7 +15,7 @@ class MainTest < IntegrationTest
 
     kamal :redeploy
     assert_app_is_up version: second_version
-    assert_hooks_ran "pre-connect", "pre-deploy", "post-deploy"
+    assert_hooks_ran "pre-connect", "pre-build", "pre-deploy", "post-deploy"
 
     assert_accumulated_assets first_version, second_version
 
@@ -44,7 +38,7 @@ class MainTest < IntegrationTest
   test "app with roles" do
     @app = "app_with_roles"
 
-    version = latest_app_version
+    version = update_app_rev
 
     assert_app_is_down