diff --git a/docling/models/layout_model.py b/docling/models/layout_model.py
index af7b8e7b..32cb1cd0 100644
--- a/docling/models/layout_model.py
+++ b/docling/models/layout_model.py
@@ -321,7 +321,7 @@ class LayoutModel:
                 clusters, page.cells, page.size.height
             )
 
-            # draw_clusters_and_cells()
+            draw_clusters_and_cells()
 
             page.predictions.layout = LayoutPrediction(clusters=clusters)
 
diff --git a/tests/test_backend_docling_parse.py b/tests/test_backend_docling_parse.py
index dd98157d..87ec2840 100644
--- a/tests/test_backend_docling_parse.py
+++ b/tests/test_backend_docling_parse.py
@@ -17,7 +17,7 @@ def test_doc_path():
 
 
 def test_get_text_from_rect(test_doc_path):
-    doc_backend = DoclingParseDocumentBackend(test_doc_path)
+    doc_backend = DoclingParseDocumentBackend(test_doc_path, "123456xyz")
     page_backend: DoclingParsePageBackend = doc_backend.load_page(0)
 
     # Get the title text of the DocLayNet paper
@@ -30,7 +30,7 @@ def test_get_text_from_rect(test_doc_path):
 
 
 def test_crop_page_image(test_doc_path):
-    doc_backend = DoclingParseDocumentBackend(test_doc_path)
+    doc_backend = DoclingParseDocumentBackend(test_doc_path, "123456xyz")
     page_backend: DoclingParsePageBackend = doc_backend.load_page(0)
 
     # Crop out "Figure 1" from the DocLayNet paper
@@ -41,5 +41,5 @@ def test_crop_page_image(test_doc_path):
 
 
 def test_num_pages(test_doc_path):
-    doc_backend = DoclingParseDocumentBackend(test_doc_path)
+    doc_backend = DoclingParseDocumentBackend(test_doc_path, "123456xyz")
     doc_backend.page_count() == 9
diff --git a/tests/test_backend_pdfium.py b/tests/test_backend_pdfium.py
index 4db7a376..56b4ee42 100644
--- a/tests/test_backend_pdfium.py
+++ b/tests/test_backend_pdfium.py
@@ -15,7 +15,7 @@ def test_doc_path():
 
 
 def test_get_text_from_rect(test_doc_path):
-    doc_backend = PyPdfiumDocumentBackend(test_doc_path)
+    doc_backend = PyPdfiumDocumentBackend(test_doc_path, "123456xyz")
     page_backend: PyPdfiumPageBackend = doc_backend.load_page(0)
 
     # Get the title text of the DocLayNet paper
@@ -28,7 +28,7 @@ def test_get_text_from_rect(test_doc_path):
 
 
 def test_crop_page_image(test_doc_path):
-    doc_backend = PyPdfiumDocumentBackend(test_doc_path)
+    doc_backend = PyPdfiumDocumentBackend(test_doc_path, "123456xyz")
     page_backend: PyPdfiumPageBackend = doc_backend.load_page(0)
 
     # Crop out "Figure 1" from the DocLayNet paper
@@ -39,5 +39,5 @@ def test_crop_page_image(test_doc_path):
 
 
 def test_num_pages(test_doc_path):
-    doc_backend = PyPdfiumDocumentBackend(test_doc_path)
+    doc_backend = PyPdfiumDocumentBackend(test_doc_path, "123456xyz")
     doc_backend.page_count() == 9