Fixed minor trace plotting bug