Fixing changing size display bug