diff --git a/Sources/Generation/Generation.swift b/Sources/Generation/Generation.swift
index 98252c6..60e3577 100644
--- a/Sources/Generation/Generation.swift
+++ b/Sources/Generation/Generation.swift
@@ -8,8 +8,9 @@
 import Foundation
 import GenerationLibrary
 
+try Generation.main()
+
 /// A type containing the generation function for the plugin.
-@main
 public enum Generation {
 
     /// Generate the Swift code for the plugin.