diff --git a/prodigy_pdf/__init__.py b/prodigy_pdf/__init__.py
index fc6fdab..7323657 100644
--- a/prodigy_pdf/__init__.py
+++ b/prodigy_pdf/__init__.py
@@ -6,7 +6,7 @@
 import pypdfium2 as pdfium
 
 from prodigy import recipe, set_hashes, ControllerComponentsDict
-from prodigy.components.stream import Stream
+from prodigy.components.stream import Stream, get_stream
 from prodigy.util import msg
 
 def page_to_image(page: pdfium.PdfPage) -> str: