Add missing import
Merge pull request #200 from mpharrigan/importfix

Add missing import