fixes issue in teee.py