diff --git a/action.yaml b/action.yaml
index 1d4c370..415dc9d 100644
--- a/action.yaml
+++ b/action.yaml
@@ -2,7 +2,7 @@ name: 'Push to Dokku'
 description: 'Pushes a repository to a remote Dokku host'
 runs:
   using: 'go'
-  main: 'main.go'
+  main: '.'
 inputs:
   host:
     description: 'Remote hostname'