fixed screen.py's import typo