Adding some real assertion in test cases