From 797799a5aa54803b1968311084c8733179d7405d Mon Sep 17 00:00:00 2001
From: rlaphoenix <rlaphoenix@pm.me>
Date: Wed, 8 Nov 2023 20:52:03 +0000
Subject: [PATCH] Slight correction to typing and doc-string of
 set_service_certificate

---
 pywidevine/cdm.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/pywidevine/cdm.py b/pywidevine/cdm.py
index 6a0159e..3448158 100644
--- a/pywidevine/cdm.py
+++ b/pywidevine/cdm.py
@@ -151,7 +151,7 @@ class Cdm:
             raise InvalidSession(f"Session identifier {session_id!r} is invalid.")
         del self.__sessions[session_id]
 
-    def set_service_certificate(self, session_id: bytes, certificate: Optional[Union[bytes, str]]) -> str:
+    def set_service_certificate(self, session_id: bytes, certificate: Optional[Union[bytes, str]]) -> Optional[str]:
         """
         Set a Service Privacy Certificate for Privacy Mode. (optional but recommended)
 
@@ -178,7 +178,8 @@ class Cdm:
                 match the underlying DrmCertificate.
 
         Returns the Service Provider ID of the verified DrmCertificate if successful.
-        If certificate is None, it will return the now unset certificate's Provider ID.
+        If certificate is None, it will return the now-unset certificate's Provider ID,
+        or None if no certificate was set yet.
         """
         session = self.__sessions.get(session_id)
         if not session: