diff --git a/src/createImports.mts b/src/createImports.mts
index 8caebaa..bcd6866 100644
--- a/src/createImports.mts
+++ b/src/createImports.mts
@@ -1,25 +1,16 @@
 import ts from "typescript";
 import { posix } from "path";
-import { Service } from "./service.mjs";
 import { Project } from "ts-morph";
 
 const { join } = posix;
 
 export const createImports = ({
-  service,
   serviceEndName,
   project,
 }: {
-  service: Service;
   serviceEndName: string;
   project: Project;
 }) => {
-  const { klasses } = service;
-  // get all class names
-  const classNames = klasses.map(({ className }) => className);
-  // remove duplicates
-  const uniqueClassNames = [...new Set(classNames)];
-
   const modelsFile = project
     .getSourceFiles()
     .find((sourceFile) => sourceFile.getFilePath().includes("models.ts"));
diff --git a/src/createSource.mts b/src/createSource.mts
index bea3f47..efa9c88 100644
--- a/src/createSource.mts
+++ b/src/createSource.mts
@@ -20,10 +20,10 @@ const createSourceFile = async (outputPath: string, serviceEndName: string) => {
   const service = await getServices(project);
 
   const imports = createImports({
-    service,
     serviceEndName,
     project,
   });
+
   const exports = createExports(service);
 
   const commonSource = ts.factory.createSourceFile(