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

adds required import.