fix the code bug