fixed the visulization polymorphism