Worked on the ROCm/aiter repository to enhance stability and maintainability by addressing a critical bug in argument parsing when the AITER_JIT_DIR environment variable is set. Applied expertise in Python, environment management, and unit testing to ensure the correct enum module loads, preventing NameError exceptions during argument checks. The solution preserved existing enum docstring cleanup behavior, improving runtime reliability and reducing test flakiness for downstream users. Delivered a concise, well-documented patch with minimal code changes, making future updates to AITER_JIT_DIR handling and enum loading more straightforward while lowering debugging time and runtime failures in affected environments.