import missing module
Merge pull request #231 from ptomulik/fix/missing-import

Fix/missing import