fix litlle bugs