diff --git a/models/roma_model.py b/models/roma_model.py
index 12536b9..7071296 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 ROMAModel(BaseModel):
+class ROMAUNSBModel(BaseModel):
     
     @staticmethod
     def modify_commandline_options(parser, is_train=True):