fixed type annotation
Merge pull request #151 from flying-sheep/patch-2

fixed type annotation