fix small plotting bug