fix code mistaken