Go to settings and then search "charging" using the search box at the top.
It should pull up an item under the Developer options "stay awake. Screen will never go to sleep when charging" in check this box.
For some reason it's defaulted and buried under the Developer options. Obviously an...