Fix the tests I added and the comment