Update models.py

When using a custom user model application fails due to wrong database relation.
This fix ensures that the proper user model is used.