Added test case
Bug fix