unit tests now works