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

adds required import.