Fixed missing yield test