diff --git a/yt_dlp_plugins/extractor/_ytse/protos/_sabr_context_update.py b/yt_dlp_plugins/extractor/_ytse/protos/_sabr_context_update.py
index 7f8e621..8f56c9e 100644
--- a/yt_dlp_plugins/extractor/_ytse/protos/_sabr_context_update.py
+++ b/yt_dlp_plugins/extractor/_ytse/protos/_sabr_context_update.py
@@ -1,6 +1,6 @@
 import typing
 import protobug
-from yt_dlp_plugins.extractor._ytse.protos import BufferedRange
+from ._buffered_range import BufferedRange
 
 
 @protobug.message
@@ -13,5 +13,5 @@ class ContextUpdate:
 class SabrContextUpdate:
     unknown_field_1: typing.Optional[protobug.Int32] = protobug.field(1, default=None)  # seen = 2
     unknown_field_2: typing.Optional[protobug.Int32] = protobug.field(2, default=None)  # seen = 2
-    context_update: ContextUpdate = protobug.field(3, default_factory=ContextUpdate),
+    context_update: typing.Optional[ContextUpdate] = protobug.field(3, default_factory=ContextUpdate),
     unknown_field_4: typing.Optional[protobug.Int32] = protobug.field(4, default=None)  # seen = 1