Adding some asserts to tests