Added one assert to test