Add missing import
Merge pull request #2 from jean/master

Add missing import oops