Condensed logic, changed toEqual to toBe