diff --git a/docs/diagrams/class_diagram.py b/docs/diagrams/class_diagram.py
index 4d6ce09365c444def48831cf52dd0c2023f7e962..21941b8ccf26685281f0f03ebc07b03f6b08265b 100644
--- a/docs/diagrams/class_diagram.py
+++ b/docs/diagrams/class_diagram.py
@@ -8,8 +8,8 @@ from pyUML import Graph, UMLClass
 
 def print_and_save(graph, name):
     #print(graph.to_string())
-    graph.write_raw(f"{name}.dot")
-    graph.write(f"{name}.png", "dot", "png")
+    #graph.write_raw(f"{name}.dot")
+    graph.write(f"{name}.png")#, "dot", "png")
 
 def generate_input_uml():
     # Create UML for the input classes