diff --git a/scripts/prepare_sector_network.py b/scripts/prepare_sector_network.py
index 8a4f98ce..7bf9f817 100644
--- a/scripts/prepare_sector_network.py
+++ b/scripts/prepare_sector_network.py
@@ -1990,7 +1990,9 @@ def add_heat(n, costs):
             if f == 0:
                 continue
             # get sector name ("residential"/"services"/or both "tot" for urban central)
-            sec = [x if x in name else "tot" for x in sectors][0]
+            if 'urban central' in name: sec = 'tot'
+            if 'residential' in name: sec = 'residential'
+            if 'services' in name: sec = 'services'
 
             # get floor aread at node and region (urban/rural) in m^2
             floor_area_node = (