diff --git a/mala/network/runner.py b/mala/network/runner.py
index a8910c2ad..ff111b10b 100644
--- a/mala/network/runner.py
+++ b/mala/network/runner.py
@@ -852,7 +852,7 @@ def _forward_entire_snapshot(
         # activated AFTER loading a model.
         if self.parameters_full.use_ddp:
             self.network.module.to(
-                self.network.params._configuration["device"]
+                self.network.module.params._configuration["device"]
             )
         else:
             self.network.to(self.network.params._configuration["device"])