Fixed the wrong attribute error