Fix type hint
Merge pull request #145 from JeroenDeDauw/th

Fix type hint