Zigbee Problems - How to get a switch module to stay connected to network?

Theres a lot to look at when troubleshooting Zigbee.

Yes repeating devices are available. Most line powered devices like your switch are in fact repeaters.

So how many other line powered zigbee devices do you have?

How far is your switch from the coordinator? (spec says one thing, walls are real, not theoretical so youll never hit the spec distance - i always assume half and then cut it in half again to get a target distance of 8-10m between repeating devices)

What Zigbee channel are you on? What channel is your 2.4 Ghz wifi using (some channels overlap)

Is there any wifi ap within a couple meters of eithe ryour coordinator or the switch?

Are there any usb3 devices within a few meters of the coordinator or switch?