Default Flashlight turns off when displays off

Hiya , I'm currently running swamp monster 4.1.2, and when I use my flashlight from the toggle. When the screen times out or goes to sleep, the flashlight kicks off until the screen comes back on..

Anyone know where in the code for the toggle to fix the screen off problem?

Sent from my SPH-L710 using XDA Free mobile app