diff --git a/mage_helpers/build.go b/mage_helpers/build.go
index c85654f977f6e05ea9ec846a069b96f6d9f38660..eb26eef9b732a1c95879927903f5630cb18d9d58 100644
--- a/mage_helpers/build.go
+++ b/mage_helpers/build.go
@@ -45,7 +45,7 @@ func Build(dir string, module string, isDebug bool) error {
 	fmt.Println(fmt.Sprintf("Building %v", handler))
 
 	outputFileName := handler
-	env := os.Getenv("ENVIRONMENT")
+	env := CurrentEnv()
 	fmt.Println("ENV:", env)
 	if env == "playground" {
 		fmt.Println("Using 'bootstrap' for output file name")