Fix type hint
Merge pull request #596 from JeroenDeDauw/typefix

Fix type hint