Fixed bug in screen driver