diff --git a/README.md b/README.md
index 333838b..90e2a0b 100644
--- a/README.md
+++ b/README.md
@@ -29,7 +29,7 @@ from aiia.model import AIIAConfig
 from aiia.pretrain import Pretrainer
 
 # Create your model
-config = AIIAConfig(model_name="AIIA-Base-512x20k")
+config = AIIAConfig(model_type="AIIA-Base-512x20k")
 model = AIIABase(config)
 
 # Initialize pretrainer with the model