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")