added more checks in test case