Fixed incorrect imports
Merge pull request #147 from alexgarbarev/master

Fixed incorrect imports