Fix the Dockerfile