Update to allow support for custom users.

Let's always use get_user_model, at this point :)