Fixing attribute error.