Fix the command for the docker image