Updated test case and removed logging code