fix screen off bug