Change InputState[S] usages for S, where S <: InputState[_]