[FIX] improve test