I had this issue. A combination of the pomodoro technique and forced daily reading made a massive difference in my short-term attention span.