Fix bug so screen can be used