Fix plotting bug.