smal bug fixed