Fix type hints