Ring Chime Pro - 3 pin socket part has come away

Hi All, My ring doorbell chime was knocked off the socket and the 3pin connector has come apart from the actual unit. It looks like a simple click back in jobby but I can't seem to do it! There are no obvious broken parts that I can see so not sure if I'm just doing it wrong? Can someone have a look at the pics and see if all looks ok? And even beffer if someone has a pic of how it should look when its together? Thank you! Saqib