diff --git a/cv7/program.c b/cv7/program.c
index 27ab8aa..0813d7a 100644
--- a/cv7/program.c
+++ b/cv7/program.c
@@ -41,6 +41,17 @@ Node* parse_knowledge_base(char lines[][MAX_LINE_LENGTH], int *index, int line_c
     return node;
 }
 
+int is_valid_tree(Node* node) {
+    if (!node) return 0;
+    if ((!node->yes && node->no) || (node->yes && !node->no)) {
+        return 0; // Некорректное состояние: одна ветка отсутствует
+    }
+    if (!node->yes && !node->no) {
+        return 1; // Листовой узел
+    }
+    return is_valid_tree(node->yes) && is_valid_tree(node->no);
+}
+
 int count_products(Node *node) {
     if (!node) return 0;
     if (!node->yes && !node->no) return 1;
@@ -110,8 +121,11 @@ int main() {
     int index = 0;
     Node *root = parse_knowledge_base(lines, &index, line_count);
 
-    if (!root) {
-        return 1;
+    if (!root || !is_valid_tree(root)) {
+        printf("Expert z bufetu to vie.\n");
+        printf("Chybna databaza\n");
+        free_tree(root);
+        return 0;
     }
 
     printf("Expert z bufetu to vie.\n");