Backlight stays on only on stock based roms

My Xperia sp's Backlight stays on when locked. This only happens I have stock rom installed. If I have a custom rom I don't have this problem. I have exactly the same apps installed whether I am on stock or custom. I prefer stock roms so would like to find a fix if possible. Thanks for your help

Sent from my C5303 using XDA Free mobile app