diff --git a/db/migrate/20201219120055_add_order_to_adjustments.rb b/db/migrate/20201219120055_add_order_to_adjustments.rb
index 85d3048fc3..9748a4372d 100644
--- a/db/migrate/20201219120055_add_order_to_adjustments.rb
+++ b/db/migrate/20201219120055_add_order_to_adjustments.rb
@@ -1,13 +1,14 @@
 class AddOrderToAdjustments < ActiveRecord::Migration
-  def up
-    class Spree::Adjustment < ActiveRecord::Base
-      belongs_to :adjustable, polymorphic: true
-      belongs_to :order, class_name: "Spree::Order"
-    end
-    class Spree::LineItem < ActiveRecord::Base
-      belongs_to :order, class_name: "Spree::Order"
-    end
+  class Spree::Adjustment < ActiveRecord::Base
+    belongs_to :adjustable, polymorphic: true
+    belongs_to :order, class_name: "Spree::Order"
+  end
 
+  class Spree::LineItem < ActiveRecord::Base
+    belongs_to :order, class_name: "Spree::Order"
+  end
+
+  def up
     add_column :spree_adjustments, :order_id, :integer
 
     # Ensure migration can use the new column