Fix visualization of CNN.