diff --git a/stable_diffusion/stable_diffusion/vae.py b/stable_diffusion/stable_diffusion/vae.py
index 1b9c8de6..fe473d4c 100644
--- a/stable_diffusion/stable_diffusion/vae.py
+++ b/stable_diffusion/stable_diffusion/vae.py
@@ -257,7 +257,7 @@ class Autoencoder(nn.Module):
 
     def __call__(self, x, key=None):
         x = self.encoder(x)
-        x = self.query_proj(x)
+        x = self.quant_proj(x)
 
         mean, logvar = x.split(2, axis=-1)
         std = mx.exp(0.5 * logvar)