Slight mod due to update plugin