fix: Fixed tests