Fix tests timeout