Fix another attribute error