Fixed another attribute error