make unit tests work