Fix build

TBR @ussuri