Update functional tests for migrate