Fix warnings

Fix warnings