diff --git a/tests/neox_args/test_neoxargs_usage.py b/tests/neox_args/test_neoxargs_usage.py
index a8cbf4707..27243c48d 100644
--- a/tests/neox_args/test_neoxargs_usage.py
+++ b/tests/neox_args/test_neoxargs_usage.py
@@ -51,6 +51,7 @@ def test_neoxargs_usage():
             "build_tokenizer",
             "attention_config[i]",
             "print",
+            "update"
         ]
     )