Minor migration hotfix