just fix small code