fixing plotting function again

2 jobs for master in 22 seconds (queued for 1 second)
Status Job ID Name Coverage
  Test
passed python_2 #440

00:00:21

passed python_3 #439

00:00:22