Oculus quest / Oculus Rift s rubber disks for battery lock
I included three different sizes.
While 2mm should be fine, depending on the TPU used it might be not enough, so there are a 3 and 4 mm versions.
Print those in any flexible filament.
Once printed, place the disk around the spring, so it does not contract as much.