Fix unit test, broken due to wording change