diff --git a/wamr-wasi-extensions/samples/nn-cli/main.c b/wamr-wasi-extensions/samples/nn-cli/main.c
index f85d647568..ca0a893ef1 100644
--- a/wamr-wasi-extensions/samples/nn-cli/main.c
+++ b/wamr-wasi-extensions/samples/nn-cli/main.c
@@ -406,7 +406,7 @@ get_output(char *options)
         exit(1);
     }
     nnret =
-        wasi_ephemeral_nn_get_output(c, 0, resultbuf, resultbufsz, &resultsz);
+        wasi_ephemeral_nn_get_output(c, idx, resultbuf, resultbufsz, &resultsz);
     if (nnret == wasi_ephemeral_nn_error_too_large) {
         resultbufsz *= 2;
         goto retry;