diff --git a/lib/userstamp/migration_helper.rb b/lib/userstamp/migration_helper.rb
index 1ff636e..4198f30 100644
--- a/lib/userstamp/migration_helper.rb
+++ b/lib/userstamp/migration_helper.rb
@@ -16,4 +16,4 @@ def userstamps(include_deleted_by = false)
   end
 end
 
-ActiveRecord::ConnectionAdapters::Table.send(:include, Ddb::Userstamp::MigrationHelper)
+ActiveRecord::ConnectionAdapters::TableDefinition.send(:include, Ddb::Userstamp::MigrationHelper)