My fix is as follows:
1. Go into developer options (if you don't know how, Google it)
2. The first option should be: 'Keep screen on when charging'. It should be disabled. Set it to Enabled.
I know it's exactly the opposite as to what we want to achieve, but just do it!
Don't let your dreams...