Fix imports
Merge pull request #107 from cgohlke/patch-5

Fix imports