diff --git a/tools/milhoja_pypkg/src/milhoja/tests/TestCodeGenerators.py b/tools/milhoja_pypkg/src/milhoja/tests/TestCodeGenerators.py
index a4698268..55a3a5ed 100644
--- a/tools/milhoja_pypkg/src/milhoja/tests/TestCodeGenerators.py
+++ b/tools/milhoja_pypkg/src/milhoja/tests/TestCodeGenerators.py
@@ -74,7 +74,7 @@ def run_tests(self, tests_all, dims_all, create_generator):
                     ref = self.__load_code(ref_hdr_fname)
                     generated = self.__load_code(header_filename)
 
-                    self.assertEqual(len(ref), len(generated))
+                    self.assertEqual(len(ref), len(generated), f"generated != {ref_hdr_fname}")
                     for gen_line, ref_line in zip(generated, ref):
                         self.assertEqual(gen_line, ref_line, f"generated != {ref_hdr_fname}")
 
@@ -95,7 +95,7 @@ def run_tests(self, tests_all, dims_all, create_generator):
                 ref = self.__load_code(ref_src_fname)
                 generated = self.__load_code(source_filename)
 
-                self.assertEqual(len(ref), len(generated))
+                self.assertEqual(len(ref), len(generated), f"generated != {ref_src_fname}")
                 for gen_line, ref_line in zip(generated, ref):
                     self.assertEqual(gen_line, ref_line, f"generated != {ref_src_fname}")