I found something that would work by searching for “NVME spacer”… not exact replacement, but Amazon sells a short hex nut with a lip on top that looks like it would do the job