Firstly, tap the blue plus button at the bottom right of the screen. This will direct you to a screen with a list of the clubs in your bag. Scroll and tap on the club that was used for the newly added shot. If your club isn’t on this list, then you will next to save your changes and add a club in the My Bag feature (soon to be realised on mobile apps) and create that club before returning to the editing feature.
Video link: https://gfycat.com/FewFalseBoutu