Fix type hints in stubs