Update the script to comply with the new outputs of the entity.