replace .get() with direct indexing in sort_stations()

1 files changed, 1 insertions, 1 deletions

diff --git a/main.py b/main.py
index 9d9217c..b6c245a 100644
--- a/main.py
+++ b/main.py
@@ -78,7 +78,7 @@ def filter_df(
 
 
 def sort_stations(stations: list[dict], sort: str) -> list[dict]:
-    sort_key = SORT_KV.get(sort)
+    sort_key = SORT_KV[sort]
     return sorted(stations, key=lambda d: d[sort_key] if d[sort_key] != "N/A" else 999)