修復了之前的BUG