fix code for the latese commit