Update for new input API.