diff --git a/cmd/api/main.go b/cmd/api/main.go
index 6ba82de..87e2ab2 100644
--- a/cmd/api/main.go
+++ b/cmd/api/main.go
@@ -1,3 +1,5 @@
+//go:build !production
+
 package main
 
 import (
@@ -7,7 +9,6 @@ import (
 	"os"
 	"time"
 
-	_ "tyapi-server/docs" // docs is generated by Swag CLI, you have to import it.
 	"tyapi-server/internal/app"
 )