[FIX] Minor test fix