Thank you for the feedback !
We were aiming to make it quite punishing to make sure the player uses light sources carefully. We were hoping to incentivize the player to explore the floor thoroughly first before the lights go out (the player could explore most of the floor for all levels, except the first one with the monster). During playtesting we also had the same thought as you but didn't have the time for further improvements sadly !