diff --git a/requirements.txt b/requirements.txt
index 9d52784..6768c62 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -32,3 +32,4 @@ scipy==1.1.0
 six==1.11.0
 tensorboard>=1.8.0
 termcolor==1.1.0
+deepaas==0.5.1
\ No newline at end of file
diff --git a/test-requirements.txt b/test-requirements.txt
index 2c6bd81..982ce07 100644
--- a/test-requirements.txt
+++ b/test-requirements.txt
@@ -14,5 +14,5 @@ pytest-cov
 
 tensorflow>=1.10.0
 pydot-ng
-deepaas
+deepaas==0.5.1
 flaat