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

Fix type hint