making unit tests work