Made test correct