fixed a todo comment in the tests