Fix #123.
Remove unnecessary import
Merge pull request #132 from lewtds/fix-issue-123

Fix issue #123