Fix linting errors.
fix unit tests again ;)