fixed a bug with the test