Add type parameter for the state.