Fix a small bug

A small bugfix, which occurs when iterating over the results for the
plot.