minor fix on migration