Minor fixes to migration object