diff --git a/.gitignore b/.gitignore
index ccad87f7..5202e40d 100755
--- a/.gitignore
+++ b/.gitignore
@@ -11,3 +11,4 @@ checkpoints/
 lightning_logs/
 pixi.lock
 .python-version
+.venv/
diff --git a/graph_weather/data/dataloader.py b/graph_weather/data/dataloader.py
index a5238d4e..28948f7b 100755
--- a/graph_weather/data/dataloader.py
+++ b/graph_weather/data/dataloader.py
@@ -117,7 +117,7 @@ def __getitem__(self, item):
             end_solar_times.append(
                 np.array([extraterrestrial_irrad(when, lat, lon) for lat, lon in lat_lons])
             )
-        end_solar_times = np.array(solar_times)
+        end_solar_times = np.array(end_solar_times)
 
         # Normalize to between -1 and 1
         solar_times -= const.SOLAR_MEAN