Fix display probleam