diff --git a/src/tests/test_norbs.py b/src/tests/test_norbs.py
index afbafd2..6be47fe 100644
--- a/src/tests/test_norbs.py
+++ b/src/tests/test_norbs.py
@@ -16,7 +16,7 @@ class TestModel(Model):
 		else:
 			# Hidden binary
 			self.unit1 = Entity((96 * 96, 333), EntityParams(do_gaussian_visible=True, do_gaussian_hidden=False),
-								TrainingParams(learning_rate=0.001, momentum=0.9, num_epochs=1000, l2_lambda=0.01))
+								TrainingParams(learning_rate=0.001, momentum=0.9, num_epochs=1000, l2_lambda=0.001))
 
 	def forward(self, x: Mat):
 		x = self.unit1.forward(x)
@@ -38,7 +38,7 @@ if __name__ == "__main__":
 	model.init(0.1)
 
 	# load state
-#	model.load()
+	model.load()
 
 	# Load train data
 	train_batch = read_armadillo(os.path.join(prj_root, f"{prj_name}.training.dat"))