lib/cm_admin.rb in cm-admin-0.7.8 vs lib/cm_admin.rb in cm-admin-0.8.0

@@ -33,10 +33,11 @@
     end
 
     def initialize_model(entity, &block)
       if entity.is_a?(Class)
         return if CmAdmin::Model.find_by({name: entity.name})
-        config.cm_admin_models << CmAdmin::Model.new(entity, &block)
+        cm_model = CmAdmin::Model.new(entity, &block)
+        config.cm_admin_models << cm_model
       end
     end
   end
 
 end