fixed bug in the edge visualization