Update to input API change.