changed type hint
Merge pull request #161 from kbond/patch-1

changed type hint