Add back accidentally deleted import.

--HG--
branch : single-codebase