fix littel bug