Apply suggestions from code review

Commit

2 years ago

Apply suggestions from code review Co-authored-by: Arthur <48595927+ArthurZucker@users.noreply.github.com>