Add back @user_count variable to make tests pass