Fixes tests/plot.rs according to changes done in plot.py

Fixes tests/plot.rs according to changes done in plot.py #642