Update migration to properly set custom user model