fix migration failed on testing