Add missing log import