Made test better