diff --git a/oi/type_graph/ClangTypeParser.cpp b/oi/type_graph/ClangTypeParser.cpp
index 5190478..2025c84 100644
--- a/oi/type_graph/ClangTypeParser.cpp
+++ b/oi/type_graph/ClangTypeParser.cpp
@@ -183,7 +183,12 @@ Type& ClangTypeParser::enumerateClass(const clang::RecordType& ty) {
 
   std::string name = decl->getNameAsString();
 
-  auto kind = Class::Kind::Struct;  // TODO: kind
+  auto kind = Class::Kind::Struct;
+  if (ty.isUnionType()) {
+    kind = Class::Kind::Union;
+  } else if (ty.isClassType()) {
+    kind = Class::Kind::Class;
+  }
 
   int virtuality = 0;