unit test added...  fixed related bug