Tweak to Dockerfile