Changed  migration file to resolve production issue