diff --git a/modneflib/modnef/modnef_torch/model.py b/modneflib/modnef/modnef_torch/model.py
index 0451bfe54d6ac45bd058e23f06f59f3e4cf78e5c..3b86faafa5ce0886c01220282104ac8b17ebf53a 100644
--- a/modneflib/modnef/modnef_torch/model.py
+++ b/modneflib/modnef/modnef_torch/model.py
@@ -181,6 +181,8 @@ class ModNEFModel(nn.Module):
       output_spike, None
     """
 
+    print("hi")
+
     def to_aer(input):
       input = input.reshape(-1).to(torch.int32)
 
@@ -197,7 +199,7 @@ class ModNEFModel(nn.Module):
     batch_result = []
 
     for sample in input_spikes:
-      sample_res = self.driver.run_sample(sample, to_aer, True, len(self.layers))
+      sample_res = self.driver.run_sample(sample, to_aer, True, 2) #TODO
       batch_result.append([sample_res])
 
     return torch.tensor(batch_result).permute(1, 0, 2), None
diff --git a/modneflib/modnef/templates/run_lib.py b/modneflib/modnef/templates/run_lib.py
index 5a987427756fa788577b2bcdaccba5224e3eea1d..07ccb6d0bdb2cc18feca1332ccf238051cf1a1b9 100644
--- a/modneflib/modnef/templates/run_lib.py
+++ b/modneflib/modnef/templates/run_lib.py
@@ -202,7 +202,7 @@ def fpga_evaluation(model, testLoader, board_path, driver_config, name="FPGA Eva
 
   return accuracy, y_pred, y_true   
 
-def confusion_matrix(y_true, y_pred, file_name, classes):
+def conf_matrix(y_true, y_pred, file_name, classes):
   cm = confusion_matrix(y_true, y_pred)
   df_cm = pd.DataFrame(cm / np.sum(cm, axis=1)[:, None], index = [i for i in classes], columns = [i for i in classes])
   plt.figure()