I don't like the `assertNotIn` check, is it possible to replace it with a `assertEquals` check.