From 10e577df971ed2a3e320ca88f3b911a8b464bf78 Mon Sep 17 00:00:00 2001
From: Leandro Pacheco <contact@leandropacheco.com>
Date: Thu, 16 Jan 2025 15:16:03 -0300
Subject: [PATCH] fix

---
 riscv-executor/src/lib.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/riscv-executor/src/lib.rs b/riscv-executor/src/lib.rs
index fac71b934..85456fa29 100644
--- a/riscv-executor/src/lib.rs
+++ b/riscv-executor/src/lib.rs
@@ -909,7 +909,7 @@ mod builder {
         }
 
         pub fn col_is_defined(&self, name: &str) -> bool {
-            if let ExecMode::Trace = self.mode {
+            if let ExecMode::Trace | ExecMode::Witness = self.mode {
                 self.trace.all_cols.contains(&name.to_string())
             } else {
                 false