fix plot (finally!)