Fixed to support custom user modules by calling get_user_model()