Fix wrong type hints, closes #1
Merge pull request #2 from Bronsa/master

Fix wrong type hints, closes #1