use list comprehension in test, fix typo