Added one extra test assertion