diff --git a/README.md b/README.md
index b282c585..14f8bc69 100644
--- a/README.md
+++ b/README.md
@@ -219,7 +219,7 @@ doesn't seem to work well.
 [28]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/recons_SWAE_49.png
 [29]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/MIWAE_29.png
 [30]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/recons_MIWAE_29.png
-[31]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/recons_VQVAE_29.png
+[31]: https://github.com/AntixK/PyTorch-VAE/blob/master/assets/recons_VQVAE_1.png
 
 [python-image]: https://img.shields.io/badge/Python-3.5-ff69b4.svg
 [python-url]: https://www.python.org/
diff --git a/assets/recons_VQVAE_1.png b/assets/recons_VQVAE_1.png
new file mode 100644
index 00000000..368f0748
Binary files /dev/null and b/assets/recons_VQVAE_1.png differ
diff --git a/models/vq_vae.py b/models/vq_vae.py
index da1c2513..bd2249c6 100644
--- a/models/vq_vae.py
+++ b/models/vq_vae.py
@@ -7,7 +7,7 @@
 class VectorQuantizer(nn.Module):
     """
     Reference:
-    [1] https://github.com/zalandoresearch/pytorch-vq-vae
+    [1] https://github.com/deepmind/sonnet/blob/v2/sonnet/src/nets/vqvae.py
     """
     def __init__(self,
                  num_embeddings: int,