Fixed a migration bug
