Small simplification in Screen code.

See also: https://github.com/prompt-toolkit/python-prompt-toolkit/pull/1877/