diff --git a/nerf_triplane/provider.py b/nerf_triplane/provider.py
index c0e1a77..b55ee4b 100644
--- a/nerf_triplane/provider.py
+++ b/nerf_triplane/provider.py
@@ -251,7 +251,7 @@ def __init__(self, opt, device, type='train', downscale=1):
         if self.opt.au45:
             import pandas as pd
             au_blink_info = pd.read_csv(os.path.join(self.root_path, 'au.csv'))
-            bs = au_blink_info[' AU45_r'].values
+            bs = au_blink_info.get('AU45_r', au_blink_info.get(' AU45_r')).values
         else:
             bs = np.load(os.path.join(self.root_path, 'bs.npy'))
             if self.opt.bs_area == "upper":