Fixed tests bug