Fixed an issue in the migration template