diff --git a/runtime_vis/vis.py b/runtime_vis/vis.py
index b397f41..14c43bc 100644
--- a/runtime_vis/vis.py
+++ b/runtime_vis/vis.py
@@ -6,7 +6,7 @@
 import scipy.optimize as opt
 import seaborn as sns
 import warnings
-from typing import Callable, Any, Iterable
+from typing import Callable, Any, Iterable, List
 
 from runtime_vis.curves import fit_n, fit_logn, fit_nlogn, fit_n2, fit_n3