fix tests (is not essential, but works)