#18 Fix test (again :( )