Code warning fix