Fix visualization problem