Add missing import
Merge branch 'master' into issue71
Merge pull request #73 from lshift/issue71

Fix missing import