fixed a small bug AttributeError