Issue #226 Fix test