Compilation warning fix