--Fix some bugs