Add missing variables import