diff --git a/Makefile b/Makefile
index 2e456df..7aaf32f 100644
--- a/Makefile
+++ b/Makefile
@@ -2,7 +2,6 @@
 # https://github.com/infobloxopen/atlas-gentool
 IMAGE_NAME := infoblox/atlas-gentool
 
-GOFLAGS                 := ""
 GO_PATH                 := /go
 SRCROOT_ON_HOST         := $(shell dirname $(abspath $(lastword $(MAKEFILE_LIST))))
 SRCROOT_IN_CONTAINER    := $(GO_PATH)/src/github.com/infobloxopen/atlas-gentool
diff --git a/go.mod b/go.mod
index f389a20..64269dc 100644
--- a/go.mod
+++ b/go.mod
@@ -17,6 +17,7 @@ require (
 	github.com/pseudomuto/protoc-gen-doc v1.5.1
 	google.golang.org/grpc/cmd/protoc-gen-go-grpc v1.2.0
 	google.golang.org/protobuf v1.33.0
+	gorm.io/gorm v1.23.2
 )
 
 require (
@@ -65,5 +66,4 @@ require (
 	google.golang.org/grpc v1.48.0 // indirect
 	gopkg.in/yaml.v2 v2.4.0 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect
-	gorm.io/gorm v1.23.2 // indirect
 )
diff --git a/tools.go b/tools.go
index 41919b9..e500dd5 100644
--- a/tools.go
+++ b/tools.go
@@ -30,4 +30,5 @@ import (
 	_ "github.com/infobloxopen/protoc-gen-gorm/errors"
 	_ "github.com/infobloxopen/protoc-gen-gorm/types"
 	_ "github.com/infobloxopen/protoc-gen-preprocess"
+	_ "gorm.io/gorm"
 )