diff --git a/test/integration/main_test.rb b/test/integration/main_test.rb
index c0c432804..28407e28b 100644
--- a/test/integration/main_test.rb
+++ b/test/integration/main_test.rb
@@ -6,17 +6,11 @@ class MainTest < IntegrationTest
     assert_env_files
     remove_local_env_file
 
-    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
@@ -25,7 +19,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
 
@@ -53,7 +47,7 @@ class MainTest < IntegrationTest
 
     kamal :envify
 
-    version = latest_app_version
+    version = update_app_rev
 
     assert_app_is_down