diff --git a/HISTORY.md b/HISTORY.md
index e8fc188..c30443a 100644
--- a/HISTORY.md
+++ b/HISTORY.md
@@ -1,3 +1,7 @@
+## 2.0.1
+
+* add missing webdataset dep
+
 ## 2.0.0
 
 * webdataset input format
diff --git a/setup.py b/setup.py
index 57b7322..c25f806 100644
--- a/setup.py
+++ b/setup.py
@@ -9,7 +9,7 @@
         name = 'clip_retrieval',
         packages = find_packages(),
         include_package_data = True,
-        version = '2.0.0',
+        version = '2.0.1',
         license='MIT',
         description = 'Easily computing clip embeddings and building a clip retrieval system with them',
         long_description=long_description,