Correct tests for new features