Corrected code to match the updated user models.