fix a bug in the test