Minor fix on imports
Merge pull request #54 from abstractj/import

Minor fix on imports