Make tests more shorter and prettier