fix code mistake