diff --git a/Sources/ResgenSwift/Generate/Generator/ArchitectureGenerator.swift b/Sources/ResgenSwift/Generate/Generator/ArchitectureGenerator.swift index ba1413c..df81bf8 100644 --- a/Sources/ResgenSwift/Generate/Generator/ArchitectureGenerator.swift +++ b/Sources/ResgenSwift/Generate/Generator/ArchitectureGenerator.swift @@ -11,12 +11,14 @@ import Foundation struct ArchitectureGenerator { static func writeArchitecture(_ architecture: ConfigurationArchitecture, projectDirectory: String) { // Create extension content - let architectureContent = [ + var architectureContent = [ "// Generated by ResgenSwift.\(Generate.toolName) \(ResgenSwiftVersion)", architecture.getClass() ] .joined(separator: "\n\n") + architectureContent += "\n" + let filename = "\(architecture.classname).swift" guard let filePath = architecture.path?.prependIfRelativePath(projectDirectory) else { let error = GenerateError.writeFile(filename, "Path of file is not defined.")