made unit tests work