diff --git a/06_gpu_and_ml/sam/segment_anything.py b/06_gpu_and_ml/sam/segment_anything.py
index 80319a6a1..1533b5369 100644
--- a/06_gpu_and_ml/sam/segment_anything.py
+++ b/06_gpu_and_ml/sam/segment_anything.py
@@ -189,7 +189,7 @@ def main(
     x_point=250,
     y_point=200,
 ):
-    with volume.batch_upload(force=True) as batch:
+    with video_vol.batch_upload(force=True) as batch:
         batch.put_file(input_video, "input.mp4")
 
     model = Model()
diff --git a/06_gpu_and_ml/stable_diffusion/image_to_image.py b/06_gpu_and_ml/stable_diffusion/image_to_image.py
index 41077c2ca..6d6c51f03 100644
--- a/06_gpu_and_ml/stable_diffusion/image_to_image.py
+++ b/06_gpu_and_ml/stable_diffusion/image_to_image.py
@@ -24,7 +24,6 @@
 from pathlib import Path
 
 import modal
-from synthetic_monitoring.benchmarks.image import CACHE_DIR
 
 CACHE_DIR = "/cache"