[Linux] remember audio stream

So I had this problem for a while now,

whenever I plug my headset into my laptop all the sounds get played via the headset. It is set to default and I tell my machine to play all audio streams over the headset. Discord follows that, except for the notification sound (and getting a call etc.), those play from the laptop speakers. When I actually manage to move the playbackstream manually to the headset it stays there until no sound is played. After that the notifications go back to the laptop speakers.

Could you make it so I can tell discord which audio device to use for notifications?