Small fix to Dockerfile