diff --git a/README.md b/README.md
index b3297084..23d002f4 100644
--- a/README.md
+++ b/README.md
@@ -34,9 +34,9 @@ So far, the following benchmarks can be used for testing.
 
 ## Results
 #### DMC Proprio
-![dmcproprio](https://github.com/NM512/dreamerv3-torch/assets/70328564/0d3f4d44-d487-4097-bed1-3f0573aaf902)
+![dmcproprio](imgs/dmcproprio.png)
 #### DMC Vision
-![dmcvision](https://github.com/NM512/dreamerv3-torch/assets/70328564/162c24d0-5f06-4ef6-bbc3-202221f6ad11)
+![dmcvision](imgs/dmcvision.png)
 #### Atari 100k
 ![atari100k](https://github.com/NM512/dreamerv3-torch/assets/70328564/0da6d899-d91d-44b4-a8c4-d5b37413aa11)
 
diff --git a/imgs/dmcproprio.png b/imgs/dmcproprio.png
new file mode 100644
index 00000000..be82e902
Binary files /dev/null and b/imgs/dmcproprio.png differ
diff --git a/imgs/dmcvision.png b/imgs/dmcvision.png
new file mode 100644
index 00000000..f84af9e9
Binary files /dev/null and b/imgs/dmcvision.png differ
diff --git a/networks.py b/networks.py
index cf9a39c9..c5d7f6a7 100644
--- a/networks.py
+++ b/networks.py
@@ -145,7 +145,6 @@ def observe(self, embed, action, is_first, state=None):
     def imagine_with_action(self, action, state):
         swap = lambda x: x.permute([1, 0] + list(range(2, len(x.shape))))
         assert isinstance(state, dict), state
-        action = action
         action = swap(action)
         prior = tools.static_scan(self.img_step, [action], state)
         prior = prior[0]