lib/scaffolding_extensions/model/datamapper.rb in scaffolding_extensions-1.4.2 vs lib/scaffolding_extensions/model/datamapper.rb in scaffolding_extensions-1.4.3

@@ -226,16 +226,16 @@
   def scaffold_save(action, object)
     object.save
   end
   
   # The column type for the given table column, or nil if it isn't a table column
-  def scaffold_table_column_type(column)
-    column = self.properties[column]
+  def scaffold_table_column_type(c)
+    column = self.properties[c]
     if column then
-      if column.type == DataMapper::Types::Text
+      if column.type == DataMapper::Property::Text
         :text
       else
-        column.type.to_s.split("::").last.downcase.intern
+        column.class.to_s.split("::").last.downcase.intern
       end
     else
       nil
     end
   end