diff --git a/element_array_ephys/probe.py b/element_array_ephys/probe.py
index b4a6d8fa..8157c022 100644
--- a/element_array_ephys/probe.py
+++ b/element_array_ephys/probe.py
@@ -1,8 +1,6 @@
 """
 Neuropixels Probes
 """
-from __future__ import annotations
-
 import datajoint as dj
 
 from .readers import probe_geometry
diff --git a/element_array_ephys/readers/probe_geometry.py b/element_array_ephys/readers/probe_geometry.py
index f9fe7fb7..11e3ae99 100644
--- a/element_array_ephys/readers/probe_geometry.py
+++ b/element_array_ephys/readers/probe_geometry.py
@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import numpy as np
 import pandas as pd