#18 Fix test