diff --git a/graph/db/migration1/test_sql.go b/graph/db/migration1/test_sql.go
index 8034bec86..3d157dc9f 100644
--- a/graph/db/migration1/test_sql.go
+++ b/graph/db/migration1/test_sql.go
@@ -31,16 +31,15 @@ func (t *testBatchedSQLQueries) ExecTx(ctx context.Context,
 	if err != nil {
 		return err
 	}
-	defer func() {
-		if err != nil {
-			_ = tx.Rollback()
-		} else {
-			err = tx.Commit()
-		}
-	}()
 
 	reset()
 	queries := sqlc.New(tx)
 
-	return txBody(queries)
+	if err := txBody(queries); err != nil {
+		_ = tx.Rollback()
+
+		return err
+	}
+
+	return tx.Commit()
 }