diff --git a/pkg/drivertest/isolation.go b/pkg/drivertest/isolation.go
index 90bdb6d..ad3f5f2 100644
--- a/pkg/drivertest/isolation.go
+++ b/pkg/drivertest/isolation.go
@@ -37,7 +37,7 @@ func testTenantIsolation1(t *testing.T, db *multitenancy.DB, opts Options) {
 			defer wg.Done()
 			tenantCtx, cancel := context.WithCancel(ctx)
 			defer cancel()
-			tx := db.WithContext(ctx)
+			tx := db.WithContext(tenantCtx)
 			if err := tx.MigrateTenantModels(tenantCtx, tenant.ID); err != nil {
 				errCh <- err
 				return