diff --git a/aidl_language.h b/aidl_language.h
index deb530a..77c8d0a 100644
--- a/aidl_language.h
+++ b/aidl_language.h
@@ -1193,6 +1193,7 @@
   AidlImport& operator=(AidlImport&&) = delete;
 
   const std::string& GetNeededClass() const { return needed_class_; }
+  std::string SimpleName() const { return needed_class_.substr(needed_class_.rfind('.') + 1); }
   void TraverseChildren(std::function<void(const AidlNode&)>) const {}
   void DispatchVisit(AidlVisitor& v) const { v.Visit(*this); }