Fixed issue with migrate