Fixed error with screen