Set proper toolClass for UGTools
optimize import