Add more asserts to test