fixed screen bug