Xiaomi X20 Pro replacement drive wheels — compatible models?

Hey everyone, I need to replace the drive wheels on my Xiaomi X20 Pro robot vacuum. On AliExpress I can only find wheels listed for the X20, X20+, and X20 Max — but not specifically for the X20 Pro.

Can anyone confirm whether any of these are actually compatible with the X20 Pro? I've seen some sellers claim the X20 Max wheel is the same, but I can't find solid confirmation anywhere.

If you've replaced yours before, feel free to drop a link to where you bought them!

Thanks!