MLDS-309 Fixing migration bug