Add unit tests to get 30% coverage