fixing plotting