Fix notification light when screen is on

Change-Id: I649c57b9ac25fb7389bcdb03711380016033ac61