Minor improve to docker image