fix StateT