Fix test where there might only be item in the list