diff --git a/database/migrations/2014_12_13_190730_changes_for_v321.php b/database/migrations/2014_12_13_190730_changes_for_v321.php
index c5a2be3f0c..84a59ab82b 100644
--- a/database/migrations/2014_12_13_190730_changes_for_v321.php
+++ b/database/migrations/2014_12_13_190730_changes_for_v321.php
@@ -1,5 +1,7 @@
 <?php
 
+use FireflyIII\Models\Budget;
+use FireflyIII\Models\Category;
 use FireflyIII\Models\BudgetLimit;
 use FireflyIII\Models\Component;
 use Illuminate\Database\Migrations\Migration;