Fix tests script