diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py
index bd371eb..b5e71ac 100644
--- a/verif/generator/tosa_test_gen.py
+++ b/verif/generator/tosa_test_gen.py
@@ -51,7 +51,11 @@
 
         fullPath = os.path.join(self.basePath, self.testPath)
         os.makedirs(fullPath, exist_ok=True)
-        self.ser = ts.TosaSerializer(fullPath, saveConstsToFile=self.args.dump_consts)
+        # Embed const data in the flatbuffer
+        constMode = ts.ConstMode.EMBED
+        if self.args.dump_consts:
+            constMode = ts.ConstMode.EMBED_DUMP
+        self.ser = ts.TosaSerializer(fullPath, constMode)
 
     def getSerializer(self):
         return self.ser