fix a bug in the tests