Fix build fallout