diff --git a/scripts/py/templates/tflite.cc.template b/scripts/py/templates/tflite.cc.template
index 97bdec5..a06c14a 100644
--- a/scripts/py/templates/tflite.cc.template
+++ b/scripts/py/templates/tflite.cc.template
@@ -16,7 +16,11 @@
 #}
 {{common_template_header}}
 
-#include "Model.hpp"
+#include "BufAttributes.hpp"
+
+#include <cstddef>
+#include <cstdint>
+
 {% for header in additional_headers %}
 #include "{{header}}"
 {% endfor %}