Made test simpler