Code cleanup: remove benchmark code from internal tests