Fix compile warning

Fix compile warning