schema compile CHANGE Notifications support
diff --git a/src/tree_schema_helpers.c b/src/tree_schema_helpers.c
index d4567dc..ef83436 100644
--- a/src/tree_schema_helpers.c
+++ b/src/tree_schema_helpers.c
@@ -1183,7 +1183,8 @@
             /* LYS_CONFIG_W, but also the default case */
             return &((struct lysc_action*)node)->input.data;
         }
-    /* TODO Notification */
+    case LYS_NOTIF:
+        return &((struct lysc_notif*)node)->data;
     default:
         return NULL;
     }