diff --git a/popv/algorithms/_onclass.py b/popv/algorithms/_onclass.py
index bc56e89..259cc43 100644
--- a/popv/algorithms/_onclass.py
+++ b/popv/algorithms/_onclass.py
@@ -179,7 +179,7 @@ def predict(self, adata):
             pred_label = [train_model.i2co[ind] for ind in onclass_seen]
             pred_label_str = [clid_2_name[ind] for ind in pred_label]
             adata.obs[self.result_key] = pred_label_str
-            adata.obs["onclass_seen"] = pred_label_str
+            adata.obs[self.seen_result_key] = pred_label_str
         else:
             onclass_pred = train_model.Predict(corr_test_feature, use_normalize=False, refine=True, unseen_ratio=-1.0)
             pred_label = [train_model.i2co[ind] for ind in onclass_pred[2]]