From 8d81e3480e29b850250d6da867cdabcb5441b53d Mon Sep 17 00:00:00 2001
From: jejuisland87654 <jejuisland87654@gmail.com>
Date: Wed, 13 Nov 2024 20:45:18 +0100
Subject: [PATCH] Fix NautilusContext::from_file for python files (#2690)

---
 libafl/src/generators/nautilus.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/libafl/src/generators/nautilus.rs b/libafl/src/generators/nautilus.rs
index 4f7d76009f..1b4b870996 100644
--- a/libafl/src/generators/nautilus.rs
+++ b/libafl/src/generators/nautilus.rs
@@ -89,9 +89,10 @@ impl NautilusContext {
         let grammar_file = grammar_file.as_ref();
         if grammar_file.extension().unwrap_or_default() == "py" {
             log::debug!("Creating NautilusContext from python grammar");
-            let ctx = python_grammar_loader::load_python_grammar(
+            let mut ctx = python_grammar_loader::load_python_grammar(
                 fs::read_to_string(grammar_file)?.as_str(),
             );
+            ctx.initialize(tree_depth);
             return Ok(Self { ctx });
         }
         log::debug!("Creating NautilusContext from json grammar");