Make models work with custom user models