fix a code bug