Fixed migration to support custom user model