Fix compile warning