Fix worm visulization