Fixed failing test after PR #152