diff --git a/modneflib/modnef/arch_builder/modules/UART/__init__.py b/modneflib/modnef/arch_builder/modules/UART/__init__.py
index 1bf960420332ba13fce617145b3d49dc5165e617..3cbbe30479f8e038dc09b9a1db57ae6c9409eabc 100644
--- a/modneflib/modnef/arch_builder/modules/UART/__init__.py
+++ b/modneflib/modnef/arch_builder/modules/UART/__init__.py
@@ -7,7 +7,6 @@ Dependencies: uart_1step, uart_classifier, uart_classifier_timer, uart_xstep, ua
 Descriptions: UART module builder init
 """
 
-from .uart_1step import Uart_1Step
 from .uart_classifier import Uart_Classifier
 from .uart_classifier_timer import Uart_Classifier_Timer
 from .uart_xstep import Uart_XStep
diff --git a/modneflib/modnef/modnef_driver/drivers/__init__.py b/modneflib/modnef/modnef_driver/drivers/__init__.py
index 0615fd915b881e40174ffa9e4a47c16cffbd1143..1bd6c7110c70072bc76e93d6790e6f290b03b768 100644
--- a/modneflib/modnef/modnef_driver/drivers/__init__.py
+++ b/modneflib/modnef/modnef_driver/drivers/__init__.py
@@ -11,6 +11,5 @@ from .classifier_driver import Classifier_Driver
 from .classifier_timer_driver import Classifier_Timer_Driver
 from .debugger_driver import Debugger_Driver
 from .default_driver import ModNEF_Driver
-from .single_step_driver import SingleStep_Driver
 from .xstep_driver import XStep_Driver
 from .xstep_timer_driver import XStep_Timer_Driver
\ No newline at end of file
diff --git a/modneflib/modnef/modnef_driver/modnef_drivers.py b/modneflib/modnef/modnef_driver/modnef_drivers.py
index be6f4db13049d828351fb97c5cb652a25359bc7f..11e2cc3226bae31637382a1f47a9c2f7fc03420f 100644
--- a/modneflib/modnef/modnef_driver/modnef_drivers.py
+++ b/modneflib/modnef/modnef_driver/modnef_drivers.py
@@ -13,7 +13,6 @@ from .drivers import *
 import yaml
 
 drivers_dict = {
-  "1Step" : SingleStep_Driver,
   "XStep" : XStep_Driver,
   "Classifier" : Classifier_Driver,
   "Debugger" : Debugger_Driver,
diff --git a/modneflib/modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py b/modneflib/modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py
index d4f58d3bd3af7fabc0ed42576a3b32c45496cab2..9bb6d1df8eeff1ab9044378d73e3625e4a9e4438 100644
--- a/modneflib/modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py
+++ b/modneflib/modnef/modnef_torch/modnef_neurons/srlif_model/shiftlif.py
@@ -269,9 +269,6 @@ class ShiftLIF(LIF, ModNEFNeuron):
     else:
       self.mem = self.mem-self.__shift(self.mem)
 
-    if self.quantization_flag:
-      self.mem.data = self.quantizer(self.mem.data, True)
-
     if self.hardware_estimation_flag:
       self.val_min = torch.min(torch.min(input_.min(), self.mem.min()), self.val_min)
       self.val_max = torch.max(torch.max(input_.max(), self.mem.max()), self.val_max)
diff --git a/modneflib/modnef/templates/run_lib.py b/modneflib/modnef/templates/run_lib.py
index 07ccb6d0bdb2cc18feca1332ccf238051cf1a1b9..21f536be3f3a60baa6bc7e834a4635c1eb9c9182 100644
--- a/modneflib/modnef/templates/run_lib.py
+++ b/modneflib/modnef/templates/run_lib.py
@@ -69,7 +69,7 @@ def train(model, trainLoader, testLoader, optimizer, loss, device=torch.device("
     acc_test_history.append(acc_test)
 
     if best_model_name!="" and acc_test>best_acc:
-      torch.save(model.state_dict(), best_model_name)
+      torch.save(model.state_dict(), f"{output_path}/{best_model_name}")
       best_acc = acc_test
 
   if save_history: