fixed a problem with the test