Add a missing import so that the code compiles again
Merge pull request #16 from benmccann/missing-import

Add a missing import so that the code compiles again