diff --git a/.github/workflows/train-spread-predictor.yaml b/.github/workflows/train-spread-predictor.yaml
index 76e96bb..7bf6696 100644
--- a/.github/workflows/train-spread-predictor.yaml
+++ b/.github/workflows/train-spread-predictor.yaml
@@ -7,7 +7,7 @@ on:
 
 jobs:
   build:
-    runs-on: ubuntu-20.04
+    runs-on: ubuntu-latest
 
     steps:
       - name: Checkout repository
diff --git a/nfl_analytics/main.py b/nfl_analytics/main.py
index 676e9a0..62098ae 100644
--- a/nfl_analytics/main.py
+++ b/nfl_analytics/main.py
@@ -71,8 +71,8 @@ def main():
     if args.train:
         start_time = time.time()
         try:
-            # df_raw = load_dataframe_from_raw()
-            df_raw = load_dataframe_from_remote()
+            df_raw = load_dataframe_from_raw()
+            # df_raw = load_dataframe_from_remote()
         except FileNotFoundError as e:
             print(f"Error loading data: {e}")
             print("Please run with --download first.")