Note10 Lite constantly turning off during games

Whenever I’m using my Note10 Lite for more strenuous activities like playing my games on it, it’ll constantly turn itself off/sleep and open immediately again with the default wallpaper flashing for a second before going back to my usual video wallpaper. It’ll only do that during these times and not when I’m on more casual apps like YouTube or social media. It all started when I updated my phone to the newest OS update on the 24th of April. Here’s the update: Notify Update