Changed tests as in example