Apply suggestion from @stevhliu

Commit

97 days ago

Apply suggestion from @stevhliu Co-authored-by: Steven Liu <59462357+stevhliu@users.noreply.github.com>