Rewrite to set WORKDIR