Use MonadState instead of StateT in testContext