diff --git a/chapter10/greedy.py b/chapter10/greedy.py
index 6f2f245..93d1003 100644
--- a/chapter10/greedy.py
+++ b/chapter10/greedy.py
@@ -12,16 +12,16 @@ stations = {
 }
 
 
-def greedy():
-    def fget():
-        needed = set()
-        for states in stations.values():
-            needed |= states
-        return needed
+def fget():
+    needed = set()
+    for states in stations.values():
+        needed |= states
+    return needed
 
+
+def greedy():
     states_needed = fget()
     logger.debug(states_needed)
-
     final_stations = set()
 
     while states_needed: