Bug fixat

