Merge pull request #41 from caglar10ur/40

add missing import, fixes #40