From 3d37194c9fecca3cd2b70e0c0d2f244454d8dc64 Mon Sep 17 00:00:00 2001
From: Antoine Pourchet <antoine.pourchet@gmail.com>
Date: Tue, 11 Dec 2018 11:40:13 -0800
Subject: [PATCH] Fixed log message for ARG step (#111)

It was previously printing "ENV" instead of "ARG".
---
 lib/builder/step/arg_step.go | 2 +-
 lib/builder/step/step.go     | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/lib/builder/step/arg_step.go b/lib/builder/step/arg_step.go
index 380d385d..adad5058 100644
--- a/lib/builder/step/arg_step.go
+++ b/lib/builder/step/arg_step.go
@@ -30,7 +30,7 @@ type ArgStep struct {
 // NewArgStep returns a BuildStep from given arguments.
 func NewArgStep(args string, name string, resolvedVal *string, commit bool) BuildStep {
 	return &ArgStep{
-		baseStep:    newBaseStep(Env, args, commit),
+		baseStep:    newBaseStep(Arg, args, commit),
 		name:        name,
 		resolvedVal: resolvedVal,
 	}
diff --git a/lib/builder/step/step.go b/lib/builder/step/step.go
index a6e0dc4e..098035cb 100644
--- a/lib/builder/step/step.go
+++ b/lib/builder/step/step.go
@@ -28,6 +28,7 @@ type Directive string
 // Set of all valid directives.
 const (
 	Add         = Directive("ADD")
+	Arg         = Directive("ARG")
 	Cmd         = Directive("CMD")
 	Copy        = Directive("COPY")
 	Entrypoint  = Directive("ENTRYPOINT")