lib/generators/petergate/install_generator.rb in petergate-0.1.4 vs lib/generators/petergate/install_generator.rb in petergate-0.1.5

@@ -12,10 +12,10 @@
         sleep 1
         Time.now.utc.strftime("%Y%m%d%H%M%S")
       end
 
       def insert_into_user_model
-        inject_into_file "app/models/user.rb", after: /^\s{2,}devise[^\n]+\n[^\n]+\n/ do
+        inject_into_file "app/models/user.rb", after: /^class\sUser < ActiveRecord::Base/ do
           <<-'RUBY'
 
   ################################################################################ 
   ## PeterGate Roles
   ## The :user role is added by default and shouldn't be included in this list.