Trivial test fix