fix a bug in TAO