lib/visual_migrate/migration.rb in visual_migrate-4.0.2.0 vs lib/visual_migrate/migration.rb in visual_migrate-4.1.0.0

@@ -1,15 +1,16 @@
 require 'active_record/migration'
 
 module VisualMigrate
   class ActiveRecord::Migrator#:nodoc:
     class << self
-      process_method = instance_method(:needs_migration?)
-      undef :needs_migration?
+      #process_method = instance_method(:needs_migration?)
+      #undef :needs_migration?
 
-      define_method :needs_migration? do
-        false#current_version < last_version
+      #define_method(needs_migration?) do |connection|
+      def needs_migration?(connection)
+        false#current_version(connection) < last_version
       end
-
     end
   end
+
 end