Fix warning

Fix warning