[FIX] fix minor bug