Fix broken ruff on main

[ArnavBalyan]

Changes

• Looks like new recent PR broke main.
• Fix the same

How I tested this

• CI

Notes

Checklist

• PR has an informative and human-readable title (this will be pulled into the release notes)
• Changes are limited to a single goal (no scope creep)
• Code passed the pre-commit check & code is left cleaner/nicer than when first encountered.
• Any change in functionality is tested
• New functions are documented (with a description, list of inputs, and expected output)
• Placeholder code is flagged / future TODOs are captured in comments
• Project documentation has been updated if adding/changing functionality.

[ArnavBalyan]

cc @skrawcz looks like a recent change caused some issues, can you pls take a look thanks!