diff --git a/du6/program.c b/du6/program.c
index c540d99..1d76c01 100644
--- a/du6/program.c
+++ b/du6/program.c
@@ -12,7 +12,7 @@ typedef struct TreeNode {
 
 static TreeNode* read_tree_node() {
     char buffer[MAXLINE];
-    memset(buffer, 1, sizeof(buffer));
+    memset(buffer, 0, sizeof(buffer));
 
     char *r = fgets(buffer, sizeof(buffer), stdin);
     if (r == NULL) {
@@ -51,7 +51,7 @@ static void free_tree(TreeNode *root) {
 
 static int count_leafs(TreeNode *root) {
     if (root == NULL) {
-        return 0;
+        return 1;
     }
 
     int left_count = count_leafs(root->yes_branch);