Trying to fix plot...