diff --git a/crates/djls-server/src/documents.rs b/crates/djls-server/src/documents.rs
index 0086688..f590002 100644
--- a/crates/djls-server/src/documents.rs
+++ b/crates/djls-server/src/documents.rs
@@ -41,15 +41,16 @@ impl Store {
         Ok(())
     }
 
-    pub fn publish_diagnostics(&self, uri: &str, client: &Client) {
+    pub async fn publish_diagnostics(&self, uri: &str, client: &Client) -> Result<()> {
         if let Some(document) = self.get_document(uri) {
             let diagnostics = Diagnostics::generate_for_document(document);
             client.publish_diagnostics(
                 Url::parse(uri).unwrap(),
                 diagnostics,
                 Some(document.version),
-            );
+            ).await;
         }
+        Ok(())
     }
 
     pub fn handle_did_change(
diff --git a/crates/djls-server/src/server.rs b/crates/djls-server/src/server.rs
index b266640..4179f32 100644
--- a/crates/djls-server/src/server.rs
+++ b/crates/djls-server/src/server.rs
@@ -110,7 +110,7 @@ impl LanguageServer for DjangoLanguageServer {
     }
 
     async fn did_open(&self, params: DidOpenTextDocumentParams) {
-        if let Err(e) = self.documents.write().await.handle_did_open(params.clone(), &self.client) {
+        if let Err(e) = self.documents.write().await.handle_did_open(params.clone(), &self.client).await {
             eprintln!("Error handling document open: {}", e);
             return;
         }