diff --git a/src/tree_schema.c b/src/tree_schema.c
index 472a1ab..7d056c9 100644
--- a/src/tree_schema.c
+++ b/src/tree_schema.c
@@ -545,7 +545,7 @@
             char *s, *id;
             const char *slash;
 
-            if ((pathtype == LYSC_PATH_DATA) && (iter->nodetype & (LYS_CHOICE | LYS_CASE))) {
+            if ((pathtype == LYSC_PATH_DATA) && (iter->nodetype & (LYS_CHOICE | LYS_CASE | LYS_INPUT | LYS_OUTPUT))) {
                 /* schema-only node */
                 continue;
             }