Fix migration for now