Fix tests on CMS