[FIX] fix bug