fixing issue with tf logging