fix small code