finally have migration working properly