Updated example tests to work with new API.