fix #80

Missing import statement
Merge pull request #988 from Borkason/fix-80

fix #80