fix transformer plot