Index: trunk/generators/activity_logger_migration/templates/migration.rb
===================================================================
--- trunk/generators/activity_logger_migration/templates/migration.rb (revision 3)
+++ trunk/generators/activity_logger_migration/templates/migration.rb (revision 44)
@@ -1,19 +1,17 @@
 class <%= class_name %> < ActiveRecord::Migration
   def self.up
     create_table :activity_logs, :options => "DEFAULT CHARSET = utf8" do |t|
-      t.column :user_id, :integer
-      t.column :activity_loggable_type, :string
-      t.column :activity_loggable_id, :integer
-      t.column :action, :string
-      t.column :created_at, :datetime
-      t.column :culprit_id, :integer
-      t.column :culprit_type, :string
-      t.column :referenced_id, :integer
-      t.column :referenced_type, :string
+      # Thanks to 'Justin' for an updated migration script, much cleaner!
+      t.belongs_to :user 
+      t.string :action 
+      t.references :activity, :null => false, :polymorphic => true 
+      t.references :culprit, :null => false, :polymorphic => true 
+      t.references :referenced, :null => false, :polymorphic => true 
+      t.timestamps 
     end
   end
 
   def self.down
     drop_table :activity_logs
   end
-end
+end
\ No newline at end of file