diff --git a/openeo/internal/jupyter.py b/openeo/internal/jupyter.py
index c117ab7b6..9fb5f94f4 100644
--- a/openeo/internal/jupyter.py
+++ b/openeo/internal/jupyter.py
@@ -47,7 +47,6 @@ def _repr_html_(self):
 
 class VisualDict(dict, JupyterIntegration):
 
-    # The first entry of the dict is expected to always be the actual dict with the data
     def __init__(self, component: str, data : dict, parameters: dict = {}):
         JupyterIntegration.__init__(self, component, data, parameters)
         dict.__init__(self, data)
@@ -55,7 +54,6 @@ def __init__(self, component: str, data : dict, parameters: dict = {}):
 
 class VisualList(list, JupyterIntegration):
 
-    # The first entry of the dict is expected to always be the actual list with the data
     def __init__(self, component: str, data : list, parameters: dict = {}):
         JupyterIntegration.__init__(self, component, data, parameters)
         list.__init__(self, data)
\ No newline at end of file
diff --git a/openeo/metadata.py b/openeo/metadata.py
index 1efd66536..212f74533 100644
--- a/openeo/metadata.py
+++ b/openeo/metadata.py
@@ -3,7 +3,7 @@
 from typing import List, Union, Tuple, Callable
 
 from openeo.util import deep_get
-from openeo.internal.jupyter import VisualDict
+from openeo.internal.jupyter import JupyterIntegration
 
 
 class MetadataException(Exception):
@@ -395,4 +395,4 @@ def add_dimension(self, name: str, label: Union[str, float], type: str = None) -
         return self._clone_and_update(dimensions=self._dimensions + [dim])
     
     def _repr_html_(self):
-        return VisualDict('collection', data = self._orig_metadata)._repr_html_()
+        return JupyterIntegration('collection', data = self._orig_metadata)._repr_html_()
diff --git a/openeo/rest/rest_capabilities.py b/openeo/rest/rest_capabilities.py
index 7924daffc..1c801dd7b 100644
--- a/openeo/rest/rest_capabilities.py
+++ b/openeo/rest/rest_capabilities.py
@@ -1,5 +1,5 @@
 from openeo.capabilities import Capabilities
-from openeo.internal.jupyter import VisualDict
+from openeo.internal.jupyter import JupyterIntegration
 
 
 class RESTCapabilities(Capabilities):
@@ -36,4 +36,4 @@ def list_plans(self):
         return self.capabilities.get('billing', {}).get('plans')
 
     def _repr_html_(self):
-        return VisualDict("capabilities", data = self.capabilities, parameters = {"url": self.url})._repr_html_()
+        return JupyterIntegration("capabilities", data = self.capabilities, parameters = {"url": self.url})._repr_html_()