Fix tests references