[FIX]: fix small issue