diff --git a/memgraph-toolbox/src/memgraph_toolbox/api/memgraph.py b/memgraph-toolbox/src/memgraph_toolbox/api/memgraph.py
index 191bad6..c9fcbe3 100644
--- a/memgraph-toolbox/src/memgraph_toolbox/api/memgraph.py
+++ b/memgraph-toolbox/src/memgraph_toolbox/api/memgraph.py
@@ -83,6 +83,7 @@ def query(self, query: str, params: dict = {}) -> List[Dict[str, Any]]:
                 parameters_=params,
                 database_=self.database,
             )
+            # TODO(gitbuda): r.data() hides Node/Edge info -> https://neo4j.com/docs/api/python-driver/current/api.html#neo4j.Record.data
             json_data = [serialize_record_data(r.data()) for r in data]
             return json_data
         except Neo4jError as e: