add migration
add migration and fix test