Make tests less brittle by simplifying the expectation