add migration
add migration & fix test