Fix BFS bugs