Remove bug from Dijkstra