lib/wcc/contentful/app.rb in wcc-contentful-app-1.0.8 vs lib/wcc/contentful/app.rb in wcc-contentful-app-1.1.0

@@ -35,10 +35,10 @@
     raise ArgumentError, 'Please first call WCC::Contentful::App.configure' if configuration.nil?
 
     WCC::Contentful.init!
 
     # Extend all model types w/ validation & extra fields
-    WCC::Contentful.types.each_value do |t|
+    WCC::Contentful::Model.schema.each_value do |t|
       file = File.dirname(__FILE__) + "/model/#{t.name.underscore}.rb"
       require file if File.exist?(file)
     end
 
     @db_connected =