修复一下这个bug。

一点都不好看，帮忙修复一下，谢谢。