Magic keyboard losing connection

Hey guys, I have the magic keyboard for iPad Pro it’s not even a year old and it has started to lag a lot when using the trackpad. It seems to lose connection for a couple of seconds as I can see it come and go in settings. When it’s plugged in via the lightning port it works fine... any ideas?

thanks