diff --git a/code/read_baseline_data.py b/code/read_baseline_data.py
index 924dfd6..6b3cd29 100644
--- a/code/read_baseline_data.py
+++ b/code/read_baseline_data.py
@@ -1,5 +1,6 @@
 import nixio as nix
 import os
+import numpy as np
 from IPython import embed
 
 def read_baseline_eod(dataset):
@@ -9,7 +10,7 @@ def read_baseline_eod(dataset):
     t = b.tags["BaselineActivity_1"]
     eod_da = b.data_arrays["LocalEOD-1"]
     eod = t.retrieve_data("LocalEOD-1")[:]
-    time = eod_da.dimensions[0].axis(len(eod))
+    time = np.asarray(eod_da.dimensions[0].axis(len(eod)))
     nix_file.close()
     return time, eod