Fix screen drawing