diff --git a/src/cogs/ollama.py b/src/cogs/ollama.py
index 8d1933f..7f90f1c 100644
--- a/src/cogs/ollama.py
+++ b/src/cogs/ollama.py
@@ -110,9 +110,11 @@ class OllamaDownloadHandler:
                     break
                 yield line
 
-    def __await__(self):
+    async def flatten(self) -> "OllamaDownloadHandler":
+        """Returns the current instance, but fully consumed."""
         async for _ in self:
             pass
+        return self
 
 
 class OllamaChatHandler: