Small tweak to User model