diff --git a/codes/models/audio/music/transformer_diffusion9.py b/codes/models/audio/music/transformer_diffusion9.py
index ce660866..e9c30ef6 100644
--- a/codes/models/audio/music/transformer_diffusion9.py
+++ b/codes/models/audio/music/transformer_diffusion9.py
@@ -299,7 +299,7 @@ def register_transformer_diffusion9(opt_net, opt):
 
 
 @register_model
-def register_transformer_diffusion8_with_quantizer(opt_net, opt):
+def register_transformer_diffusion9_with_quantizer(opt_net, opt):
     return TransformerDiffusionWithQuantizer(**opt_net['kwargs'])