[FIX] Broken tests