Revert "fix plot bench"

This reverts commit 609239b71ad814f92e8564d4e640456c8f180f1c.