Worked on the radiocosmology/draco repository to improve the reliability of the Flagging Module by addressing a critical bug in the ApplyGenericMask task. Focused on Python-based code refactoring, the developer identified and corrected a typo that previously caused incorrect mask axis attribute access, which had the potential to disrupt data processing workflows. The fix restored proper handling of mask axes, reducing the risk of downstream analysis errors and enhancing the robustness of the masking workflow. All changes were implemented with clear commit documentation for traceability, demonstrating attention to maintainability and auditability in collaborative scientific software development environments.