Litle bugg fix