It may have nothing to do with the loss of connection, but which stepper motors did you get with the WorkBee?

If you bought the 2.8A (Duet2 WiFi upper limit) versions did you change the 2.4A limit to 2.8A.

I went with the Mesa Electronics 5i25, the Geckodrive G540, and LinuxCNC for my WorkBee.