Trying to fix #2

Trying to fix that only half of screen works.

See #2.