Small bug fix with z-index