Adding other test to migration_test.rb