Adding checks in test cases