diff --git a/intermediate_source/coding_ddpg.py b/intermediate_source/coding_ddpg.py
index c54736bf1f..12e8b82bba 100644
--- a/intermediate_source/coding_ddpg.py
+++ b/intermediate_source/coding_ddpg.py
@@ -768,7 +768,7 @@ def make_ddpg_actor(
     ).to(device)
 
     # initialize lazy modules
-    qnet(actor(proof_environment.reset()))
+    qnet(actor(proof_environment.reset().to(device)))
     return actor, qnet