Add missing use statement
Merge pull request #160 from johanobergman/patch-1

Add missing use statement