Added check for category icons