Fix tests to handle list-component