Thanks for pointing this out. I have confused the size of output with the size of state.