diff --git a/Sources/SourceGraph/Mutators/EntryPointAttributeRetainer.swift b/Sources/SourceGraph/Mutators/EntryPointAttributeRetainer.swift
index 6b1a5dc15..9e4ced96e 100644
--- a/Sources/SourceGraph/Mutators/EntryPointAttributeRetainer.swift
+++ b/Sources/SourceGraph/Mutators/EntryPointAttributeRetainer.swift
@@ -10,7 +10,7 @@ final class EntryPointAttributeRetainer: SourceGraphMutator {
 
     func mutate() {
         graph
-            .declarations(ofKinds: [.class, .struct])
+            .declarations(ofKinds: [.class, .struct, .enum])
             .lazy
             .filter {
                 $0.attributes.contains("NSApplicationMain") ||