BUG: Fix display bug