fix class not found issue