Minor bug-fix in screen size handling