fix the docker image build