fix tests (whoops)