diff --git a/crates/ai2/src/prompts/repository_context.rs b/crates/ai2/src/prompts/repository_context.rs
index 78db5a1651..1bb75de7d2 100644
--- a/crates/ai2/src/prompts/repository_context.rs
+++ b/crates/ai2/src/prompts/repository_context.rs
@@ -2,7 +2,7 @@ use crate::prompts::base::{PromptArguments, PromptTemplate};
 use std::fmt::Write;
 use std::{ops::Range, path::PathBuf};
 
-use gpui2::{AsyncAppContext, Handle};
+use gpui2::{AsyncAppContext, Model};
 use language2::{Anchor, Buffer};
 
 #[derive(Clone)]
@@ -14,7 +14,7 @@ pub struct PromptCodeSnippet {
 
 impl PromptCodeSnippet {
     pub fn new(
-        buffer: Handle<Buffer>,
+        buffer: Model<Buffer>,
         range: Range<Anchor>,
         cx: &mut AsyncAppContext,
     ) -> anyhow::Result<Self> {