diff --git a/cmd/apply/apply_integration_test.go b/cmd/apply/apply_integration_test.go
index 77e03861..4e07a531 100644
--- a/cmd/apply/apply_integration_test.go
+++ b/cmd/apply/apply_integration_test.go
@@ -30,11 +30,7 @@ func TestMain(m *testing.M) {
 	sharedEmbeddedPG = testutil.SetupPostgres(nil)
 	defer sharedEmbeddedPG.Stop()
 
-	// Run tests
-	code := m.Run()
-
-	// Exit with test result code
-	os.Exit(code)
+	m.Run()
 }
 
 // TestApplyCommand_TransactionRollback verifies that the apply command uses proper
diff --git a/cmd/migrate_integration_test.go b/cmd/migrate_integration_test.go
index de4d2210..778c9acf 100644
--- a/cmd/migrate_integration_test.go
+++ b/cmd/migrate_integration_test.go
@@ -37,11 +37,7 @@ func TestMain(m *testing.M) {
 	sharedEmbeddedPG = testutil.SetupPostgres(nil)
 	defer sharedEmbeddedPG.Stop()
 
-	// Run tests
-	code := m.Run()
-
-	// Exit with test result code
-	os.Exit(code)
+	m.Run()
 }
 
 // TestPlanAndApply tests the complete CLI (plan and apply) workflow using test cases
diff --git a/internal/diff/diff_test.go b/internal/diff/diff_test.go
index 9ca88841..b1386d5a 100644
--- a/internal/diff/diff_test.go
+++ b/internal/diff/diff_test.go
@@ -19,11 +19,7 @@ func TestMain(m *testing.M) {
 	sharedTestPostgres = testutil.SetupPostgres(nil)
 	defer sharedTestPostgres.Stop()
 
-	// Run tests
-	code := m.Run()
-
-	// Exit with test result code
-	os.Exit(code)
+	m.Run()
 }
 
 // buildSQLFromSteps builds a SQL string from collected plan diffs
diff --git a/internal/plan/plan_test.go b/internal/plan/plan_test.go
index 7b2d1239..628af849 100644
--- a/internal/plan/plan_test.go
+++ b/internal/plan/plan_test.go
@@ -26,11 +26,7 @@ func TestMain(m *testing.M) {
 	sharedTestPostgres = testutil.SetupPostgres(nil)
 	defer sharedTestPostgres.Stop()
 
-	// Run tests
-	code := m.Run()
-
-	// Exit with test result code
-	os.Exit(code)
+	m.Run()
 }
 
 // discoverTestDataVersions discovers available test data versions in the testdata directory