Fixing migration
Maybe a fix now...hopefully