Trying to fix back the migration