diff --git a/cmake/AddHEIR.cmake b/cmake/AddHEIR.cmake
index 043fcd56c..2f0efa3c0 100644
--- a/cmake/AddHEIR.cmake
+++ b/cmake/AddHEIR.cmake
@@ -20,8 +20,8 @@ function(add_heir_dialect dialect dialect_namespace)
   # Ops
   if(EXISTS "${CMAKE_CURRENT_SOURCE_DIR}/${dialect}Ops.td")
     set(LLVM_TARGET_DEFINITIONS ${dialect}Ops.td)
-    mlir_tablegen(${dialect}Ops.h.inc -gen-op-decls -dialect=lwe)
-    mlir_tablegen(${dialect}Ops.cpp.inc -gen-op-defs -dialect=lwe)
+    mlir_tablegen(${dialect}Ops.h.inc -gen-op-decls -dialect=${dialect_namespace})
+    mlir_tablegen(${dialect}Ops.cpp.inc -gen-op-defs -dialect=${dialect_namespace})
     add_public_tablegen_target(HEIR${dialect}OpsIncGen)
     add_dependencies(HEIR${dialect}IncGen HEIR${dialect}OpsIncGen)
   endif()