Added asserts to tests