diff --git a/makefile b/makefile
index e03682f..1a92386 100644
--- a/makefile
+++ b/makefile
@@ -1,8 +1,8 @@
 example:
-	go run ./src/main.go ./examples/in/ ./examples/out/
+	go run ./src/main.go ./examples/in/ ./examples/out/ testpackage
 
 example_container:
 	docker run --pull always --rm -v "${PWD}/examples:/examples" gitea.linuxcode.net/linuxcode/oapi-type-definitions-extractor:latest /examples/in /examples/out 
 
 lint:
-	docker run --rm -v "${PWD}/src:/src"  golangci/golangci-lint:v1.50 /bin/sh -c "cd /src && golangci-lint run"
\ No newline at end of file
+	docker run --rm -v "${PWD}/src:/src"  golangci/golangci-lint:v1.50 /bin/sh -c "cd /src && golangci-lint run"
diff --git a/src/main.go b/src/main.go
index 0aa195f..8159bd5 100644
--- a/src/main.go
+++ b/src/main.go
@@ -11,8 +11,8 @@ import (
 )
 
 func main() {
-	if len(os.Args) != 3 {
-		fmt.Println("Usage: go run main.go <input_directory> <output_directory>")
+	if len(os.Args) != 4 {
+		fmt.Println("Usage: go run main.go <input_directory> <output_directory> <package_name>")
 		return
 	}