more unit tests (found a bug!)