Improve User model test