Add missing import
Merge pull request #1182 from jfirebaugh/imports

Add missing import