Fix build break (reviewed with Raghu in person)