fix code bug