間違って削除していた処理を復活。

なぜ必要なのか忘れてるが、こうでないと駄目のよう。
コンパイル警告を出ないよう修正