Fixed smal bug