Correct example for new API changes