diff --git a/tests/model_outputs_from_annotation_test.py b/tests/model_outputs_from_annotation_test.py
index 5e002fc..fa4a96e 100644
--- a/tests/model_outputs_from_annotation_test.py
+++ b/tests/model_outputs_from_annotation_test.py
@@ -12,8 +12,8 @@ if __name__ == '__main__':
     ]
     # generate random predictions
     [
-        vis_com.generate_random_prediction()
+        vis_com.from_random()
         for vis_com in vis_com_list
-    ]  # type: ignore
+    ]
     for vis_com in vis_com_list:
         print(vis_com)