Fix class div wrapper