Fix a bugs