Minor warning fix