[FIX] Broken test