Fix build-breaking change