Fixed some visualization