fix class not found issues
fix class not found issue