fix fogotten code