a little modification

修改了一些bug