diff --git a/scaaml/capture/input_generators/attack_point_iterator.py b/scaaml/capture/input_generators/attack_point_iterator.py
index 1baf6697..fd8821ad 100644
--- a/scaaml/capture/input_generators/attack_point_iterator.py
+++ b/scaaml/capture/input_generators/attack_point_iterator.py
@@ -17,7 +17,7 @@
 """
 
 from abc import ABC, abstractmethod
-from typing import Dict, List
+from typing import List
 
 
 class AttackPointIterator: