lib/pickle/adapters/mongoid.rb in pickle-0.4.3 vs lib/pickle/adapters/mongoid.rb in pickle-0.4.4

@@ -15,10 +15,10 @@
         ObjectSpace.each_object(Class).to_a.select {|klass| klass.ancestors.include? Mongoid::Document}
       end
 
       # get a list of column names for a given class
       def self.column_names(klass)
-        klass.fields.keys
+        klass.try(:fields).try(:keys) || []
       end
 
       # Get an instance by id of the model
       def self.get_model(klass, id)
         klass.find(id)