small fix on migration tool after integration tests