Fix attribute mapping