Made test example more general