re-fix plot bug