Fix imports.
Merge pull request #26 from Dumbo52/master

Fix imports.