Compile warning fix