Fix display: none issue