Fix build breakage