minor fix
fix migration