Fix missing import.

Reported by Zero Cho.

Reviewed at https://rbcommons.com/s/twitter/r/1483/
Sync of https://github.com/pantsbuild/pants/commit/9fab5d4604fc6332d890800016ba6c5f8826b630