diff --git a/examples/IF_sbi.py b/examples/IF_sbi.py
index b7b8ec0..8de5adc 100644
--- a/examples/IF_sbi.py
+++ b/examples/IF_sbi.py
@@ -57,9 +57,9 @@ def n_peaks(x):
 inferencer = Inferencer(dt=dt, model=eqs_inf,
                         input={'I_syn': inp_trace.reshape(1, -1)},
                         output={'v': out_trace},
-                        features=[lambda x: x[(t > start_syn) & (t < end_syn), :].mean(axis=1),
-                                  lambda x: x[(t > start_syn) & (t < end_syn), :].std(axis=1),
-                                  lambda x: x[(t > start_syn) & (t < end_syn), :].max(axis=1),
+                        features=[lambda x: x[(t > start_syn) & (t < end_syn), :].mean(axis=0),
+                                  lambda x: x[(t > start_syn) & (t < end_syn), :].std(axis=0),
+                                  lambda x: x[(t > start_syn) & (t < end_syn), :].max(axis=0),
                                   n_peaks],
                         method='exponential_euler',
                         threshold='v > -50 * mV',