Add missing test for User model