diff --git a/meerkat/interactive/app/src/lib/component/contrib/flash_fill/__init__.py b/meerkat/interactive/app/src/lib/component/contrib/flash_fill/__init__.py
index 9733b45f9..0cd965e8d 100644
--- a/meerkat/interactive/app/src/lib/component/contrib/flash_fill/__init__.py
+++ b/meerkat/interactive/app/src/lib/component/contrib/flash_fill/__init__.py
@@ -25,8 +25,6 @@ def __init__(
         manifest_cache_dir: str = "~/.cache/manifest",
         max_tokens: int = 1,
     ):
-        self.max_tokens = max_tokens
-
         df = df.view()
         if target_column not in df.columns:
             df[target_column] = ""
@@ -41,6 +39,7 @@ def __init__(
             os.path.expanduser(manifest_cache_dir)
         )
         os.makedirs(self.manifest_cache_dir, exist_ok=True)
+        self.max_tokens = max_tokens
 
     @property
     def prompt(self):