Fix code broken