diff --git a/modneflib/modnef/modnef_torch/model.py b/modneflib/modnef/modnef_torch/model.py
index 3b86faafa5ce0886c01220282104ac8b17ebf53a..bf63ae0a948894ad047788c1173bf4de8d5d171e 100644
--- a/modneflib/modnef/modnef_torch/model.py
+++ b/modneflib/modnef/modnef_torch/model.py
@@ -162,11 +162,11 @@ class ModNEFModel(nn.Module):
     """
 
     if self.fpga_eval_flag:
-      return self._fpga_forward(input_spikes)
+      return self.fpga_forward(input_spikes)
     else:
       return self.software_forward(input_spikes)
   
-  def _fpga_forward(self, input_spikes):
+  def fpga_forward(self, input_spikes):
     """
     Transmit input spike to FPGA
 
@@ -181,8 +181,6 @@ class ModNEFModel(nn.Module):
       output_spike, None
     """
 
-    print("hi")
-
     def to_aer(input):
       input = input.reshape(-1).to(torch.int32)