Fix visualization error