Fix tests build