fixing the migration tasks