diff --git a/runtime/core/bin/kws_main.cc b/runtime/core/bin/kws_main.cc
index 298a139..16e5bb5 100644
--- a/runtime/core/bin/kws_main.cc
+++ b/runtime/core/bin/kws_main.cc
@@ -52,7 +52,7 @@ int main(int argc, char* argv[]) {
     for (int i = 0; i < prob.size(); i++) {
       std::cout << "frame " << offset + i << " prob";
       for (int j = 0; j < prob[i].size(); j++) {
-        std::cout << " " << prob[i][0];
+        std::cout << " " << prob[i][j];
       }
       std::cout << std::endl;
     }