Bug fix while creating migration file