[FIX] Fix test logics