Tiny update on User model