fix and improve cosmetic error

I ve tested, and it work. Do check if i miss anything