From 44076782dcb86af3309fe2bda909d971f9819266 Mon Sep 17 00:00:00 2001
From: Thinh Nguyen <thinh@vathes.com>
Date: Wed, 13 Sep 2023 18:02:31 -0500
Subject: [PATCH] bugfix

---
 element_array_ephys/probe.py                  | 2 --
 element_array_ephys/readers/probe_geometry.py | 2 ++
 2 files changed, 2 insertions(+), 2 deletions(-)

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