Fix attribute errors