diff --git a/pkg/storage/unified/sql/test/integration_test.go b/pkg/storage/unified/sql/test/integration_test.go
index 99c3d17b77a..90a3a54d635 100644
--- a/pkg/storage/unified/sql/test/integration_test.go
+++ b/pkg/storage/unified/sql/test/integration_test.go
@@ -49,7 +49,7 @@ func TestIntegrationStorageServer(t *testing.T) {
 		})
 		require.NoError(t, err)
 		require.NotNil(t, backend)
-		err = backend.Init(testutil.NewDefaultTestContext(t))
+		err = backend.Init(testutil.NewTestContext(t, time.Now().Add(1*time.Minute)))
 		require.NoError(t, err)
 		return backend
 	})
@@ -72,7 +72,7 @@ func TestIntegrationSQLStorageBackend(t *testing.T) {
 			})
 			require.NoError(t, err)
 			require.NotNil(t, backend)
-			err = backend.Init(testutil.NewDefaultTestContext(t))
+			err = backend.Init(testutil.NewTestContext(t, time.Now().Add(1*time.Minute)))
 			require.NoError(t, err)
 			return backend
 		}, nil)
@@ -91,7 +91,7 @@ func TestIntegrationSQLStorageBackend(t *testing.T) {
 			})
 			require.NoError(t, err)
 			require.NotNil(t, backend)
-			err = backend.Init(testutil.NewDefaultTestContext(t))
+			err = backend.Init(testutil.NewTestContext(t, time.Now().Add(1*time.Minute)))
 			require.NoError(t, err)
 			return backend
 		}, nil)