diff --git a/lib/userstamp/migration_helper.rb b/lib/userstamp/migration_helper.rb index 1ff636e..6bebf8c 100644 --- a/lib/userstamp/migration_helper.rb +++ b/lib/userstamp/migration_helper.rb @@ -16,4 +16,6 @@ def userstamps(include_deleted_by = false) end end +# ActiveRecord::ConnectionAdapters::Table.send(:include, Ddb::Userstamp::MigrationHelper) +ActiveRecord::ConnectionAdapters::TableDefinition.send(:include, Ddb::Userstamp::MigrationHelper) ActiveRecord::ConnectionAdapters::Table.send(:include, Ddb::Userstamp::MigrationHelper) diff --git a/lib/userstamp/stampable.rb b/lib/userstamp/stampable.rb index f76fdd7..fb44889 100644 --- a/lib/userstamp/stampable.rb +++ b/lib/userstamp/stampable.rb @@ -111,7 +111,7 @@ def without_stamps end def stamper_class #:nodoc: - stamper_class_name.to_s.capitalize.constantize rescue nil + stamper_class_name.to_s.camelize.constantize rescue nil end end @@ -148,4 +148,5 @@ def set_deleter_attribute end end -ActiveRecord::Base.send(:include, Ddb::Userstamp::Stampable) if defined?(ActiveRecord) \ No newline at end of file +ActiveRecord::Base.send(:include, Ddb::Userstamp::Stampable) if defined?(ActiveRecord) +