fixed smal bug