fixing plotting function again

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

00:00:21

passed python_3 #439

00:00:22