Fix migrate.rb