Fix visualization