Minor: added asserts in test case