Liten bugg fix