Modify test for user model