Xiaomi Button Delay

I have just setup a Xiaomi Gateway and Mijia Wireless Switch.

I’m using the button to change the brightness of my bed side lamp. When i press the button there is a delay of about a second before the light changes, this I can deal with.

The problem seems to be as if the button or gateway is going to sleep and the first button press takes several seconds to register, just long enough to make you think it hasn’t worked and press it again.

Does anybody else who used Xiaomi kit see a delay in the same way?

Is there anything I can do like sending some form of heartbeat to keep everything awake?

Thanks.