Apply suggestion from @stevhliu

Commit

171 days ago

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