diff --git a/dnn/dred_rdovae.c b/dnn/dred_rdovae.c
index 89a4eb9fb704117f7446f8d95010379b7e671c7e..74a97ec1950a0580c0ac0dfbd0b22a9168f8928b 100644
--- a/dnn/dred_rdovae.c
+++ b/dnn/dred_rdovae.c
@@ -33,7 +33,7 @@ RDOVAEEnc * DRED_rdovae_create_encoder()
     return enc;
 }
 
-RDOVAEEnc * DRED_rdovae_create_decoder()
+RDOVAEDec * DRED_rdovae_create_decoder()
 {
     RDOVAEDec *dec;
     dec = (RDOVAEDec*) calloc(sizeof(*dec), 1);
diff --git a/dnn/include/dred_rdovae.h b/dnn/include/dred_rdovae.h
index cb22bfdd5ab676db43266972bca501d981da1d68..3e781a0d7536e0cd2cd5339e46c26eff93189ab7 100644
--- a/dnn/include/dred_rdovae.h
+++ b/dnn/include/dred_rdovae.h
@@ -8,7 +8,7 @@ size_t DRED_rdovae_get_enc_size(void);
 
 size_t DRED_rdovae_get_dec_size(void);
 
-RDOVAEEnc * DRED_rdovae_create_decoder(void);
+RDOVAEDec * DRED_rdovae_create_decoder(void);
 
 RDOVAEEnc * DRED_rdovae_create_encoder(void);