Fix test migration