diff --git a/vars/main.yml b/vars/main.yml
index f93926f..c6d9ec7 100644
--- a/vars/main.yml
+++ b/vars/main.yml
@@ -1,8 +1,7 @@
 ---
 py3_env_install: True
 py3_env_pip_pkgs:
-  - docker-compose
-  - docker
+  - docker-py
   - jsondiff
 
 docker_deb_repo_key_url: 'https://download.docker.com/linux/ubuntu/gpg'