fix 2 tests