Fix unit tests :-/