diff --git a/kenning/optimizers/tvm.py b/kenning/optimizers/tvm.py
index 69c24c40..d60e5d3a 100644
--- a/kenning/optimizers/tvm.py
+++ b/kenning/optimizers/tvm.py
@@ -671,6 +671,9 @@ def generator():
                 with open(outputpath.with_suffix(".c"), "w") as ops_f:
                     ops_f.write(lib.get_lib().get_source())
 
+                    if not self.zephyr_template_header:
+                        return None
+
                     # extract TVM functions from source file
                     tvm_funcs = list(
                         set(