Added checks to test case