Fix pep8 error in bench tool

pep8 tests were failing because of a nit in the imports.

Change-Id: Ib7fc8afaa2c81266d41072d2da7d104f2ed9c87d
Closes-bug: #1366110
Merge "Fix pep8 error in bench tool"