Fixing display none issue