-

diff --git a/src/docker-wrapper b/src/docker-wrapper
index 9c78334..802f8b7 100755
--- a/src/docker-wrapper
+++ b/src/docker-wrapper
@@ -2,7 +2,7 @@
 export DOCKER_ENV="$SNAP/docker-env"
 export PATH="$DOCKER_ENV/usr/sbin:$DOCKER_ENV/usr/bin:$DOCKER_ENV/sbin:$DOCKER_ENV/bin:$PATH"
 export LD_LIBRARY_PATH="$DOCKER_ENV/lib:$DOCKER_ENV/usr/lib:$DOCKER_ENV/lib/x86_64-linux-gnu:$DOCKER_ENV/usr/lib/x86_64-linux-gnu:$LD_LIBRARY_PATH"
-export PYTHONPATH="$DOCKER_ENV/lib/python2.7/site-packages"
+export PYTHONPATH="$DOCKER_ENV/lib/python2.7/site-packages:$DOCKER_ENV/lib/python2.7/site-packages/yaml"
 
 if [ "$1" = 'docker-compose' ]; then
     cmd="$1"