adds required import.
Merge pull request #34 from birkin/master

adds required import.