Fix tests JS