diff --git a/stac_ipyleaflet/widgets/draw.py b/stac_ipyleaflet/widgets/draw.py
index 350ab99..7c55d7c 100644
--- a/stac_ipyleaflet/widgets/draw.py
+++ b/stac_ipyleaflet/widgets/draw.py
@@ -12,6 +12,7 @@ def template(self, **kwargs) -> Box(style={"max_height: 200px"}):
         bbox_out = Output()
 
         # Set unwanted draw controls to False or empty objects
+        # @TODO-CLEANUP: Create only one DrawControl and pass in the attributes instead
         draw_control = DrawControl(
             edit=False,
             remove=False,
diff --git a/stac_ipyleaflet/widgets/inspect.py b/stac_ipyleaflet/widgets/inspect.py
index 26ed75d..e0def0e 100644
--- a/stac_ipyleaflet/widgets/inspect.py
+++ b/stac_ipyleaflet/widgets/inspect.py
@@ -21,6 +21,8 @@ def template(
         self, applied_layers, interact_widget, make_get_request, titiler_endpoint
     ):
         main = self
+        
+        # @TODO-CLEANUP: Create only one DrawControl and pass in the attributes instead
         draw_control = DrawControl(
             edit=False,
             remove=False,