lib/generators/koudoku/install_generator.rb in koudoku-1.1.1 vs lib/generators/koudoku/install_generator.rb in koudoku-1.1.2

@@ -45,10 +45,10 @@
       # Add coupons.
       generate("model coupon code:string free_trial_length:string")
       template "app/models/coupon.rb"
 
       # Update the owner relationship.
-      inject_into_class "app/models/#{subscription_owner_model}.rb", Plan,
+      inject_into_class "app/models/#{subscription_owner_model}.rb", subscription_owner_model.camelize.constantize,
                         "# Added by Koudoku.\n  has_one :subscription\n\n"
 
       # Install the pricing table.
       copy_file "app/views/koudoku/subscriptions/_social_proof.html.erb"