diff --git a/lib/generators/github/ds/active_record_generator.rb b/lib/generators/github/ds/active_record_generator.rb
index 45e86e7..1d631fb 100644
--- a/lib/generators/github/ds/active_record_generator.rb
+++ b/lib/generators/github/ds/active_record_generator.rb
@@ -10,7 +10,7 @@ class ActiveRecordGenerator < ::Rails::Generators::Base
         source_paths << File.join(File.dirname(__FILE__), "templates")
 
         def create_migration_file
-          migration_template "migration.rb", "db/migrate/create_key_values_table.rb", migration_version: migration_version
+          migration_template "migration.rb.tt", "db/migrate/create_key_values_table.rb", migration_version: migration_version
         end
 
         def self.next_migration_number(dirname)
diff --git a/lib/generators/github/ds/templates/migration.rb b/lib/generators/github/ds/templates/migration.rb.tt
similarity index 78%
rename from lib/generators/github/ds/templates/migration.rb
rename to lib/generators/github/ds/templates/migration.rb.tt
index 079e95c..aa30840 100644
--- a/lib/generators/github/ds/templates/migration.rb
+++ b/lib/generators/github/ds/templates/migration.rb.tt
@@ -9,11 +9,9 @@ def self.up
 
     add_index  <%= table_name %>, :key, :unique => true
     add_index  <%= table_name %>, :expires_at
-
-    change_column  <%= table_name %>, :id, "bigint(20) NOT NULL AUTO_INCREMENT"
   end
 
   def self.down
-    drop_table  <%= table_name %>
+    drop_table <%= table_name %>
   end
 end
diff --git a/test/generators/github/store/active_record_generator_test.rb b/test/generators/github/store/active_record_generator_test.rb
index de4d64b..bf7efe3 100644
--- a/test/generators/github/store/active_record_generator_test.rb
+++ b/test/generators/github/store/active_record_generator_test.rb
@@ -29,12 +29,10 @@ def self.up
 
     add_index  :#{table_name}, :key, :unique => true
     add_index  :#{table_name}, :expires_at
-
-    change_column  :#{table_name}, :id, "bigint(20) NOT NULL AUTO_INCREMENT"
   end
 
   def self.down
-    drop_table  :#{table_name}
+    drop_table :#{table_name}
   end
 end
 EOM