Fix issue with plotting all results