作文 高校生 3年弱前

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