Add missing 'use' statement