diff --git a/template.yml b/template.yml
index 2dd7627..401ef80 100644
--- a/template.yml
+++ b/template.yml
@@ -13,7 +13,7 @@ Resources:
       FunctionName: legalgo-api
       Handler: main
       Runtime: go1.x
-      CodeUri: .
+      CodeUri: cmd/legalgo
       Events:
         Api:
           Type: Api