diff --git a/tox.ini b/tox.ini
index f043dd4..038acc1 100644
--- a/tox.ini
+++ b/tox.ini
@@ -12,6 +12,7 @@ deps =
     flake8-comprehensions
     flake8-logging-format
     flake8-mutable
+    importlib-metadata==4.13.0
 commands = flake8 daft/ train.py ablation_adni_survival.py ablation_adni_classification.py
 
 [testenv:black]