Lock screen fix