ci | Subhas/Viky | Fix small issues with migration model