Fix test for new list