Fixed display issue