Fixed the tests to do with breadth first search