Missing import
Merge pull request #2 from andybak/patch-1

Missing import