diff --git a/neuro_py/tuning/fields.py b/neuro_py/tuning/fields.py
index 4c214a2..ad4f080 100644
--- a/neuro_py/tuning/fields.py
+++ b/neuro_py/tuning/fields.py
@@ -1,5 +1,5 @@
 from math import sqrt
-from typing import Any, Callable, List, Optional, Tuple
+from typing import Any, Callable, Dict, List, Optional, Tuple
 
 import matplotlib.pyplot as plt
 import numpy as np