Possibly fix transformation plots disappearing