Fixed another migration problems