Changed return type in addCustomAttribute method