small change to make the screen thing work