Fixed a bug in the test

Screen size wasn't getting set properly