Add missing import.
Merge pull request #4 from jasondavies/fix-build

Add missing import.