Fix failing CI build.
Fix failing CI build.