diff --git a/firefly/base_encoder.py b/firefly/base_encoder.py
index 57e35e7..1b6ef46 100644
--- a/firefly/base_encoder.py
+++ b/firefly/base_encoder.py
@@ -1,7 +1,9 @@
 import abc
 import torch
 
-from typing import Tuple
+from typing import List
+
+from firefly.frame_extractor.frame import VideoFrame, AudioFrame
 
 class BaseVisionEncoder(metaclass=abc.ABCMeta):
     @abc.abstractmethod
diff --git a/mypy.ini b/mypy.ini
index 6ed21c0..a261c1f 100644
--- a/mypy.ini
+++ b/mypy.ini
@@ -17,4 +17,7 @@ ignore_missing_imports = True
 ignore_missing_imports = True
 
 [mypy-msclap.*]
+ignore_missing_imports = True
+
+[mypy-timm.*]
 ignore_missing_imports = True
\ No newline at end of file