Fixed AttributeError due to old name