Tried to fix yield test