Adding in missing 'use' statement