diff --git a/figures/Schmidt2018_dyn/compute_granger_causality.py b/figures/Schmidt2018_dyn/compute_granger_causality.py
index e4575826e9bf083a7fc50a8da1fbd1b923e062cd..6d725aba9abc10e35cbee45b1f52d9c0b511d469 100644
--- a/figures/Schmidt2018_dyn/compute_granger_causality.py
+++ b/figures/Schmidt2018_dyn/compute_granger_causality.py
@@ -49,7 +49,7 @@ connection_params = {'g': -11.,
 network_params = {'connection_params': connection_params}
 M = MultiAreaModel(network_params)
 # We exclude external input from the analysis
-K = M.K_matrix[:, :-1]
+K = M.K[:, :-1]
 
 
 def indices_to_population(structure, indices):
@@ -150,7 +150,7 @@ else:
             test = (np.nan, np.nan)
             res_red = np.nan
 
-            gc[source_area][source_pop] = (cause, test[1])
+        gc[source_area][source_pop] = (cause, test[1])
 
 fn = os.path.join(save_path,
                   'granger_causality_{}_{}.json'.format(area, pop))