Fix migration.. omg