smal bug fix