diff --git a/whampy/dataReduction.py b/whampy/dataReduction.py
index f733a7e..c22d902 100644
--- a/whampy/dataReduction.py
+++ b/whampy/dataReduction.py
@@ -11,6 +11,8 @@
 import glob
 import os.path
 
+from astropy.table import Table
+
 from scipy.interpolate import interp1d
 from scipy.optimize import minimize
 from unittest import mock