code bug fixed