diff --git a/server/main_test.go b/server/main_test.go
index 4e828f3..d951d31 100644
--- a/server/main_test.go
+++ b/server/main_test.go
@@ -87,6 +87,12 @@ func TestMain(t *testing.T) {
 
 	loc := time.Now().Location()
 
+	AssertCategoriesAndAvailableEqual(t, loc, handler, ctx, budget)
+
+	// check accounts
+}
+
+func AssertCategoriesAndAvailableEqual(t *testing.T, loc *time.Location, handler *Handler, ctx context.Context, budget *postgres.Budget) {
 	resultDir := "../testdata/production-export/results"
 	files, err := os.ReadDir(resultDir)
 	if err != nil {
@@ -108,10 +114,6 @@ func TestMain(t *testing.T) {
 		testCaseFile := filepath.Join(resultDir, file.Name())
 		handler.CheckAvailableBalance(ctx, t, testCaseFile, budget, first)
 	}
-
-	// check categories
-
-	// check accounts
 }
 
 type TestData struct {