diff --git a/figures/MAM2EBRAINS.py b/figures/MAM2EBRAINS.py
index 4c4b4404db0b6e4ac5d63104eb2c8a9c95ff3866..ce9208f41dce297b2d71980ecbd13f1271ac62ad 100644
--- a/figures/MAM2EBRAINS.py
+++ b/figures/MAM2EBRAINS.py
@@ -11,10 +11,13 @@ from matplotlib import gridspec
 from matplotlib import rc_file
 rc_file('plotstyle.rc')
 
+icolor = myred
+ecolor = myblue
+
 # Instantaneous and mean firing rate across all populations
 
-def plot_instan_mean_firing _rate(tsteps, rate):
-    fig, ax = plt.subplots()
+def plot_instan_mean_firing_rate(tsteps, rate, sim_params):
+    fig, ax = pl.subplots()
     ax.plot(tsteps, rate)
     ax.plot(tsteps, np.average(rate)*np.ones(len(tsteps)), label='mean')
     ax.set_title('Instantaneous and mean firing rate across all populations')
diff --git a/multi-area-model.ipynb b/multi-area-model.ipynb
index c0b1814ab273d7dcdb89d5b07e61c2a28baee492..b38965b55d245b48b3e319d7fad42c3f0b0aae59 100644
--- a/multi-area-model.ipynb
+++ b/multi-area-model.ipynb
@@ -1113,7 +1113,7 @@
     }
    ],
    "source": [
-    "M2E.plot_instan_mean_firing _rate(tsteps, rate)"
+    "M2E.plot_instan_mean_firing _rate(tsteps, rate, sim_params)"
    ]
   },
   {