modify dijkstra test code