Build warning fix