minor plot fix