fix some bugs
small change