fixes migration and test