diff --git a/README.md b/README.md
index 320908e..c1e4639 100644
--- a/README.md
+++ b/README.md
@@ -25,10 +25,10 @@ Pre-trained representations are becoming crucial for many NLP and perception tas
 from PIL import Image
 import requests
 
-from transformers import ALIGNProcessor, ALIGNModel
+from transformers import AlignProcessor, AlignModel
 
-model = ALIGNModel.from_pretrained("kakaobrain/align-base")
-processor = ALIGNProcessor.from_pretrained("kakaobrain/align-base")
+processor = AlignProcessor.from_pretrained("kakaobrain/align-base")
+model = AlignModel.from_pretrained("kakaobrain/align-base")
 
 url = "http://images.cocodataset.org/val2017/000000039769.jpg"
 image = Image.open(requests.get(url, stream=True).raw)