added missing import
Merge pull request #12 from nirbir/patch-2

added missing import