small fix on migration