diff --git a/builder.go b/builder.go
index 6c389b34b4..5ede4ae5b2 100644
--- a/builder.go
+++ b/builder.go
@@ -235,6 +235,9 @@ func (builder *Builder) Build(dockerfile io.Reader, stdout io.Writer) (*Image, e
 				return nil, err
 			}
 			fmt.Fprintf(stdout, "===> %s\n", base.ShortId())
+
+			image = base
+
 			break
 		default:
 			fmt.Fprintf(stdout, "Skipping unknown op %s\n", tmp[0])