diff --git a/src/cli.c b/src/cli.c
index 55115d6..fef34e9 100644
--- a/src/cli.c
+++ b/src/cli.c
@@ -464,9 +464,10 @@
 		return 0;
 
 	appctx->io_handler = kw->io_handler;
+	appctx->io_release = kw->io_release;
+	/* kw->parse could set its own io_handler or ip_release handler */
 	if ((!kw->parse || kw->parse(args, appctx, kw->private) == 0) && appctx->io_handler) {
 		appctx->st0 = CLI_ST_CALLBACK;
-		appctx->io_release = kw->io_release;
 	}
 	return 1;
 }