diff --git a/tests/wrappers/bio_ner_predictor_test.py b/tests/wrappers/bio_ner_predictor_test.py
index 5017278..a94cd34 100644
--- a/tests/wrappers/bio_ner_predictor_test.py
+++ b/tests/wrappers/bio_ner_predictor_test.py
@@ -102,7 +102,7 @@ def setUp(self):
         self.set_up_succeed = True
 
     def test_predict(self):
-        if set_up_succeed:
+        if self.set_up_succeed:
             for idx, data_pack in enumerate(self.pl.process_dataset(self.input_path)):
                 ems = list(data_pack.get_data(EntityMention))
                 self.assertTrue(len(ems) > 0)