Fix display none and animation