From e2e901b01e1a374caf86cad92bbc1d722a4402f5 Mon Sep 17 00:00:00 2001
From: bjhardcastle <ben.hardcastle@alleninstitute.org>
Date: Thu, 14 Sep 2023 00:32:19 -0700
Subject: [PATCH] Fix type

---
 src/npc_sessions/sessions.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/npc_sessions/sessions.py b/src/npc_sessions/sessions.py
index a03824a4..ff51835c 100644
--- a/src/npc_sessions/sessions.py
+++ b/src/npc_sessions/sessions.py
@@ -963,7 +963,7 @@ def stim_frame_times(self) -> dict[str, npt.NDArray[np.float64]]:
             }
         asserted_stim_frame_times: dict[str, npt.NDArray[np.float64]] = {}
         for k, v in self._stim_frame_times.items():
-            utils.assert_stim_times(self._stim_frame_times[k])
+            v = utils.assert_stim_times(v)
             asserted_stim_frame_times[k] = v
         assert not any(isinstance(v, Exception) for v in asserted_stim_frame_times.values())
         return asserted_stim_frame_times