diff --git a/tae.hpp b/tae.hpp
index 68305982..c458b87d 100644
--- a/tae.hpp
+++ b/tae.hpp
@@ -201,7 +201,7 @@ struct TinyAutoEncoder : public GGMLRunner {
                     bool decoder_only = true,
                     SDVersion version = VERSION_SD1)
         : decode_only(decoder_only),
-          taesd(decode_only, version),
+          taesd(decoder_only, version),
           GGMLRunner(backend) {
         taesd.init(params_ctx, tensor_types, prefix);
     }