diff --git a/figures/MAM2EBRAINS.py b/figures/MAM2EBRAINS.py
index afa764b144fa4a2c657aae83a8df41c5266c156f..922e10c4f80d54bcd4b8f4b50fc97bfad33c09a7 100644
--- a/figures/MAM2EBRAINS.py
+++ b/figures/MAM2EBRAINS.py
@@ -18,7 +18,7 @@ ecolor = myblue
 
 # Instantaneous and mean firing rate across all populations
 def plot_instan_mean_firing_rate(tsteps, rate, sim_params):
-    fig, ax = pl.subplots()
+    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')
@@ -67,7 +67,7 @@ def set_boxplot_props(d):
     pl.setp(d['means'], marker='x', color='k',
             markerfacecolor='k', markeredgecolor='k', markersize=3.)
 
-def plot_resting_state():   
+def plot_resting_state(A):   
     """
     Figure layout
     """