Create module to clarify what code is just for migrating from files.

1 files changed, 2 insertions, 2 deletions

diff --git a/app/controllers/admin_public_body_category_controller.rb b/app/controllers/admin_public_body_category_controller.rb
index 9380aa21b..99a4a7f3c 100644
--- a/app/controllers/admin_public_body_category_controller.rb
+++ b/app/controllers/admin_public_body_category_controller.rb
@@ -42,7 +42,7 @@ class AdminPublicBodyCategoryController < AdminController
                     end
 
                     added_headings.each do |heading_id|
-                        @category.add_to_heading(PublicBodyHeading.find(heading_id))
+                        PublicBodyHeading.find(heading_id).add_category(@category)
                     end
                 end
 
@@ -83,7 +83,7 @@ class AdminPublicBodyCategoryController < AdminController
             if @category.save
                 if params[:headings]
                     params[:headings].values.each do |heading_id|
-                        @category.add_to_heading(PublicBodyHeading.find(heading_id))
+                        PublicBodyHeading.find(heading_id).add_category(@category)
                     end
                 end
                 flash[:notice] = 'Category was successfully created.'