Fixed #2 AttributeError Error