ご返信ありがとうございます。
そのように修正しましたらエラーが出なくなりました!
また、5.0.3で修正を反映していただけたようでありがとうございました!
ソースを見比べたところ、5.0.0までは plugins_loadedフックの時点で読み込みが完了しているため、テーマの読み込みの時点ですでに動作していましたが、5.0.2では initフックの際に読み込むようになっており、テーマの読み込み時点では動作しなくなっていました。
以前のようにテーマの読み込み時点で動作するように挙動を戻してもらえませんでしょうか?