fix compile error in example

the example did not compile for me without the missing import
Merge pull request #11 from tomer-ben-david/master

fix compile error in example