diff --git a/chains-examples/docs/audio-transcription/transcribe.py b/chains-examples/docs/audio-transcription/transcribe.py
index 53733d9f2..2dfceed0d 100644
--- a/chains-examples/docs/audio-transcription/transcribe.py
+++ b/chains-examples/docs/audio-transcription/transcribe.py
@@ -24,8 +24,8 @@ class DeployedWhisper(chains.StubBase):
     async def run_remote(
         self, whisper_input: data_types.WhisperInput
     ) -> data_types.WhisperResult:
-        resp = await self._remote.predict_async(
-            json_payload={"whisper_input": whisper_input.model_dump()},
+        resp = await self.predict_async(
+            inputs={"whisper_input": whisper_input.model_dump()},
         )
         return data_types.WhisperResult.parse_obj(resp)
 
diff --git a/chains-examples/docs/audio-transcription/whisper_chainlet.py b/chains-examples/docs/audio-transcription/whisper_chainlet.py
index c67fa5bad..c12dd605b 100644
--- a/chains-examples/docs/audio-transcription/whisper_chainlet.py
+++ b/chains-examples/docs/audio-transcription/whisper_chainlet.py
@@ -17,7 +17,9 @@ def base64_to_wav(base64_string, output_file_path):
 class WhisperModel(chains.ChainletBase):
     remote_config = chains.RemoteConfig(
         docker_image=chains.DockerImage(
-            base_image="baseten/truss-server-base:3.10-gpu-v0.9.0",
+            base_image=chains.CustomImage(
+                image="baseten/truss-server-base:3.10-gpu-v0.9.0"
+            ),
             apt_requirements=[
                 "ffmpeg",
             ],