Update migration to prevent error.