Fixed another migration bug