diff --git a/Makefile b/Makefile
index 856fa25..5645faf 100644
--- a/Makefile
+++ b/Makefile
@@ -57,7 +57,7 @@ ${WORKSPACE}/traceability_agent:
 				-X 'github.com/Axway/agent-sdk/pkg/cmd.SDKBuildVersion=$(sdk_version)' \
 				-X 'github.com/Axway/agent-sdk/pkg/cmd.BuildAgentName=KongTraceabilityAgent' \
 				-X 'github.com/Axway/agent-sdk/pkg/cmd.BuildAgentDescription=Kong Traceability Agent'" \
-		-a -o ${WORKSPACE}/bin/traceability_agent ${WORKSPACE}/traceability/main.go
+		-a -o ${WORKSPACE}/bin/traceability_agent ${WORKSPACE}/pkg/main/traceability/main.go
 
 build-ta: dep ${WORKSPACE}/traceability_agent
 	@echo "Traceability Agent build completed"