simple fix; +2 tests