diff --git a/assets/INSTALL.md b/assets/INSTALL.md
index 045809c..941b9b3 100644
--- a/assets/INSTALL.md
+++ b/assets/INSTALL.md
@@ -7,7 +7,7 @@ We test the codes in the following environments, other versions may also be comp
 - Pytorch 2.1.0
 - Torchvison 0.16.0
 
-## Install environment for UNINEXT
+## Install environment for GLEE
 
 ```
 pip3 install shapely==1.7.1
diff --git a/assets/images/imagetask.png b/assets/images/imagetask.png
index bd45484..d1b0a92 100644
Binary files a/assets/images/imagetask.png and b/assets/images/imagetask.png differ
diff --git a/projects/GLEE/glee/GLEE.py b/projects/GLEE/glee/GLEE.py
index a98970f..0d315be 100644
--- a/projects/GLEE/glee/GLEE.py
+++ b/projects/GLEE/glee/GLEE.py
@@ -380,7 +380,7 @@ class GLEE(nn.Module):
                 self.inference_rvos(batched_inputs, prompt_list, task)
                 return 
             elif task in ['ytbvos']:
-                self.inference_ytbvos_uninext(batched_inputs, prompt_list, task)
+                self.inference_ytbvos(batched_inputs, prompt_list, task)
             elif task in ['omnilabel']:
                 return self.omnilabel_inference(batched_inputs, task)
             else: