Essay Senior High over 3 yearsago

Wordに「電車が来るのを待って」とうったら、文法的ミスがあると指摘されました。 「電車が来ることを待って」と書いた方がいいんですか？ また、この文の良い言い換えなどがあったら教えて欲しいです✨