more unit tests!