diff --git a/requirements.txt b/requirements.txt
index d554115..a46df8e 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -2,5 +2,5 @@ apache-airflow[s3]>=1.8.1
 joblib==0.11
 alembic==0.9.8
 
-skills-ml ~= 0.2
+skills-ml==2.0.0
 skills-utils ~= 0.2