fixing unit tests again