Minor changes in migration template