fix unit tests, how embarassing