Fixed #7

Detail:
- Corrected type hinting