diff --git a/golang_build.py b/golang_build.py
index dc4f0c6..43d5734 100644
--- a/golang_build.py
+++ b/golang_build.py
@@ -139,7 +139,9 @@ def run(self, task='build', flags=None):
                 flags = new_flags
 
             if not found_filename:
-                flags.append(self.window.active_view().file_name())
+                for file in os.listdir(working_dir):
+                    if file.endswith(".go"):
+                        flags.append(working_dir + "/" + file)
 
         if task == 'cross_compile':
             _task_cross_compile(