Added additional assert in test case