This requires setting the correct StartupWMClass=... line in the .desktop file. Unfortunately I forget how exactly to figure out what to put for the ... but I think the method I used before was something like: with the application open, run this in Terminal
xprop | grep -P -i 'wm.?class'
… then click the application’s window and look at the output? I think this won’t work in Wayland though 