Fix import in a python3 compatible way

Remove a relative path
Merge pull request #207 from Kitton/patch-1

Fix import in a python3 compatible way