fix compile error in example

the example did not compile for me without the missing import