Fixed tests failure