diff --git a/src/coverup/segment.py b/src/coverup/segment.py
index e4dd6b5..fa3b679 100644
--- a/src/coverup/segment.py
+++ b/src/coverup/segment.py
@@ -95,6 +95,7 @@ def get_imports(n):
     while names:
         name = names.pop()
         if imp := imap.get(name):
+            # TODO limit 'import' to names actually needed
             imports.append(ast.unparse(imp[1]))
             names -= imp[0]