lib/schema_comments/migration.rb in schema_comments-0.4.0 vs lib/schema_comments/migration.rb in schema_comments-0.4.1

@@ -1,18 +1,14 @@
 module SchemaComments
   module Migration
-    def self.included(mod)
-      mod.extend(ClassMethods)
-      mod.instance_eval do
-        alias :migrate_without_schema_comments :migrate
-        alias :migrate :migrate_with_schema_comments
-      end
+    def self.prepended(mod)
+      mod.singleton_class.prepend(ClassMethods)
     end
 
     module ClassMethods
-      def migrate_with_schema_comments(*args, &block)
+      def migrate(*args, &block)
         SchemaComments::SchemaComment.yaml_access do
-          migrate_without_schema_comments(*args, &block)
+          super(*args, &block)
         end
       end
     end
 
   end