diff --git a/lib/tasks/sample_data/payment_method_factory.rb b/lib/tasks/sample_data/payment_method_factory.rb
index 4febf09fa4..ba545b2d11 100644
--- a/lib/tasks/sample_data/payment_method_factory.rb
+++ b/lib/tasks/sample_data/payment_method_factory.rb
@@ -53,8 +53,8 @@ module SampleData
         environment: Rails.env,
         distributor_ids: [enterprise.id]
       )
-      calculator.calculable = payment_method
-      calculator.save!
+      payment_method.calculator = calculator
+      payment_method.save!
     end
   end
 end