diff --git a/setup.py b/setup.py
index d19d11c60c..d250ace479 100644
--- a/setup.py
+++ b/setup.py
@@ -44,6 +44,7 @@
         "gym_sokoban@git+https://github.com/Learning-and-Intelligent-Systems/gym-sokoban.git",  # pylint: disable=line-too-long
         "pbrspot@git+https://github.com/NishanthJKumar/pbrspot.git",
         "google-generativeai",
+        "imagehash",
         "rich",
     ],
     include_package_data=True,