Fix chain for StateT