Add unit test to cover the interface