Fix a warning in the test by adding the List type