Fixes display bug