Fix tests build error