Fix tests compilation