Fix build breakage from previous commit