[Photon S] Looking for Photon S Z-axis stepper motor/lead screw replacement

I'm looking to buy a new stepper motor (and lead screw? I think they're essentially a unit) for my Photon S. I can't find many options but can find plenty listed for the original Photon (like this from FEPShop, for example https://fepshop.com/shop/parts/anycubic-photon-stepper-motor-with-leadscrew/), are these easily compatible? I know when I was doing some research a while ago there was talk about a collar that the original had which doesn't exist on the Photon S or something of the sort, so I can't tell if these are two different parts.

​

Thanks!