diff --git a/capellambse_context_diagrams/collectors/class_tree.py b/capellambse_context_diagrams/collectors/class_tree.py
index 9190f1fb..3cf26934 100644
--- a/capellambse_context_diagrams/collectors/class_tree.py
+++ b/capellambse_context_diagrams/collectors/class_tree.py
@@ -83,8 +83,7 @@ def _process_box(
     ) -> None:
         if obj.uuid not in self.made_boxes:
             self.made_boxes.add(obj.uuid)
-            box = self._make_box(obj, partition, params)
-            self._set_data_types_and_labels(box, obj)
+            self._make_box(obj, partition, params)
 
     def _make_box(
         self, obj: information.Class, partition: int, params: dict[str, t.Any]