diff --git a/hubconf.py b/hubconf.py
index eb26625..c7e5b7f 100644
--- a/hubconf.py
+++ b/hubconf.py
@@ -1,6 +1,5 @@
 dependencies = ['torch', 'torchaudio', 'numpy']
 import torch
-from omegaconf import OmegaConf
 from utils import (init_jit_model,
                    get_speech_ts,
                    save_audio,
diff --git a/utils.py b/utils.py
index a9edce6..da360df 100644
--- a/utils.py
+++ b/utils.py
@@ -248,7 +248,7 @@ def single_audio_stream(model,
         yield states
 
 
-def collect_speeches(tss: List(dict),
+def collect_speeches(tss: List[dict],
                      wav: torch.Tensor):
     speech_chunks = []
     for i in tss: