diff --git a/pytket/extensions/cutensornet/general_state/tensor_network_state.py b/pytket/extensions/cutensornet/general_state/tensor_network_state.py
index 8123b81e..65ef3c27 100644
--- a/pytket/extensions/cutensornet/general_state/tensor_network_state.py
+++ b/pytket/extensions/cutensornet/general_state/tensor_network_state.py
@@ -36,7 +36,7 @@
     warnings.warn("local settings failed to import cuquantum", ImportWarning)
 
 
-class GeneralState:  # TODO: Write it as a context manager so that I can call free()
+class GeneralState:
     """Wrapper of cuTensorNet's NetworkState for exact simulation of states.
 
     Note:
@@ -331,7 +331,7 @@ def __exit__(self, exc_type: Any, exc_value: Any, exc_tb: Any) -> None:
         self.destroy()
 
 
-class GeneralBraOpKet:  # TODO: Write it as a context manager
+class GeneralBraOpKet:
     """Wrapper of cuTensorNet's NetworkState for exact simulation of ``<bra|O|ket>``.
 
     Note: