Fixed failing test after #152