Fix wrong type hints, closes #1