Fix compile warning problem