Add missing headers for e2e test image build Upstream-commit: ee21669f5329733cced55e1c678cd6cee393a6e4 Component: engine