Fix image build