small changes to tee api