diff --git a/Cigarette/Alg/alg_jd_yolo8.cpp b/Cigarette/Alg/alg_jd_yolo8.cpp
index 912b3887..3e173db0 100644
--- a/Cigarette/Alg/alg_jd_yolo8.cpp
+++ b/Cigarette/Alg/alg_jd_yolo8.cpp
@@ -106,6 +106,7 @@ bool YOLO_V8::InitParam(DL_INIT_PARAM& iParams)
     cudaEnable = iParams.cudaEnable;
     intraOpNumThreads = iParams.intraOpNumThreads;
     logSeverityLevel = iParams.logSeverityLevel;
+    classes = { "jd" };
     return true;
 }
 
@@ -224,7 +225,7 @@ char* YOLO_V8::RunSession(cv::Mat& in, cv::Mat& out, std::vector<std::pair<int,
 
         float confidence = floor(100 * re.confidence) / 100;
         std::cout << std::fixed << std::setprecision(2);
-        std::string label = std::string("jd") + std::string(" ") +
+        std::string label = classes[re.classId] + std::string(" ") +
             std::to_string(confidence).substr(0, std::to_string(confidence).size() - 4);
 
         cv::rectangle(