diff --git a/figures/Schmidt2018/helpers.py b/figures/Schmidt2018/helpers.py
index 927ed22b5b564b482b8520cc5a05ef58da254364..1c2de1285bfe0086d1a9768babd1c00b555dadbe 100644
--- a/figures/Schmidt2018/helpers.py
+++ b/figures/Schmidt2018/helpers.py
@@ -61,7 +61,7 @@ def hierarchical_relation(target_area, source_area, SLN_completed, thresh=(0.35,
         return 'same-area'
 
 
-def structural_gradient(target_area, source_area, structure):
+def structural_gradient(target_area, source_area, arch_types):
     """
     Returns the structural gradient between two areas
     See Schmidt, M., Bakker, R., Hilgetag, C.C. et al.
@@ -74,11 +74,13 @@ def structural_gradient(target_area, source_area, structure):
         Name of target area.
     source_area : str
         Name of source area.
+    arch_types : dict
+       Dictionary containing the architectural type for each area.
     """
     if target_area != source_area:
-        if structure[target_area] < structure[source_area]:
+        if arch_types[target_area] < arch_types[source_area]:
             return 'HL'
-        elif structure[target_area] > structure[source_area]:
+        elif arch_types[target_area] > arch_types[source_area]:
             return 'LH'
         else:
             return 'HZ'