Fixed the migration error