diff --git a/src/decentralizepy/node/Node.py b/src/decentralizepy/node/Node.py
index 3e0ffbabe6fd00050c4d1aa3e73b05df5f608b92..77c29d97fabe6f6fed614ea27a7ba641177bad3b 100644
--- a/src/decentralizepy/node/Node.py
+++ b/src/decentralizepy/node/Node.py
@@ -159,6 +159,7 @@ class Node:
             self.trainer.train(self.dataset)
 
             self.sharing.step()
+            self.optimizer = optimizer_class(self.model.parameters(), **optimizer_params) # Reset optimizer state
 
             rounds_to_test -= 1