Changed assertion in example test