fix unit test, was too fragile.