trivial import fixes

Hi there,
Thanks for the samples!
There are currently some broken imports in that module.
Cheers,-C