Assert proper return types from state calls.

Just to be sure.