Fixing viz error