Updated tests for migrate