diff --git a/platform/internal/handlers/workspace_budget_test.go b/platform/internal/handlers/workspace_budget_test.go
index 554816cc..97a54e2a 100644
--- a/platform/internal/handlers/workspace_budget_test.go
+++ b/platform/internal/handlers/workspace_budget_test.go
@@ -137,6 +137,7 @@ func TestWorkspaceBudget_Create_WithLimit(t *testing.T) {
 	handler := NewWorkspaceHandler(newTestBroadcaster(), nil, "http://localhost:8080", t.TempDir())
 
 	budgetVal := int64(1000) // $10.00
+	mock.ExpectBegin()
 	mock.ExpectExec("INSERT INTO workspaces").
 		WithArgs(
 			sqlmock.AnyArg(), // id
@@ -148,9 +149,10 @@ func TestWorkspaceBudget_Create_WithLimit(t *testing.T) {
 			(*string)(nil),   // parent_id
 			nil,              // workspace_dir
 			"none",           // workspace_access
-			&budgetVal,       // budget_limit
+			&budgetVal,       // budget_limit ($10)
 		).
 		WillReturnResult(sqlmock.NewResult(0, 1))
+	mock.ExpectCommit()
 	mock.ExpectExec("INSERT INTO canvas_layouts").
 		WithArgs(sqlmock.AnyArg(), float64(0), float64(0)).
 		WillReturnResult(sqlmock.NewResult(0, 1))
diff --git a/platform/internal/handlers/workspace_test.go b/platform/internal/handlers/workspace_test.go
index 56bc5ebb..b524d412 100644
--- a/platform/internal/handlers/workspace_test.go
+++ b/platform/internal/handlers/workspace_test.go
@@ -234,7 +234,7 @@ func TestWorkspaceCreate_WithSecrets_Persists(t *testing.T) {
 
 	mock.ExpectBegin()
 	mock.ExpectExec("INSERT INTO workspaces").
-		WithArgs(sqlmock.AnyArg(), "Hermes Agent", nil, 1, "hermes", sqlmock.AnyArg(), (*string)(nil), nil, "none").
+		WithArgs(sqlmock.AnyArg(), "Hermes Agent", nil, 1, "hermes", sqlmock.AnyArg(), (*string)(nil), nil, "none", (*int64)(nil)).
 		WillReturnResult(sqlmock.NewResult(0, 1))
 	// Secret inserted inside the same transaction.
 	mock.ExpectExec("INSERT INTO workspace_secrets").
diff --git a/platform/migrations/025_workspace_budget.down.sql b/platform/migrations/027_workspace_budget.down.sql
similarity index 100%
rename from platform/migrations/025_workspace_budget.down.sql
rename to platform/migrations/027_workspace_budget.down.sql
diff --git a/platform/migrations/025_workspace_budget.up.sql b/platform/migrations/027_workspace_budget.up.sql
similarity index 100%
rename from platform/migrations/025_workspace_budget.up.sql
rename to platform/migrations/027_workspace_budget.up.sql