diff --git a/models/roma_model.py b/models/roma_model.py
index 7071296..12536b9 100644
--- a/models/roma_model.py
+++ b/models/roma_model.py
@@ -15,7 +15,7 @@ import math
 
 from torchvision.transforms import transforms as tfs
 
-class ROMAUNSBModel(BaseModel):
+class ROMAModel(BaseModel):
     
     @staticmethod
     def modify_commandline_options(parser, is_train=True):