Fixed bug on visualization