fix some codes
fix codes