diff --git a/lib/cloud_builder/cli.rb b/lib/cloud_builder/cli.rb
index ee51bb2..64d556d 100644
--- a/lib/cloud_builder/cli.rb
+++ b/lib/cloud_builder/cli.rb
@@ -33,7 +33,7 @@ def execute
         end
       end
 
-      if diff?
+      if (diff? || update?)
         remote_template = cf.stacks[key].template
         
         # b = s3.buckets[bucket]
@@ -51,7 +51,10 @@ def execute
         cmd = "%s %s %s" % [diff_tool ? diff_tool : "git diff --color", t1.path, t2.path]
         # puts cmd
         puts `#{cmd}`
-        return
+        if (diff? && !update?)
+          return
+        end
+        #return
       end
 
       if bucket
@@ -76,4 +79,4 @@ def execute
 
     end
   end
-end
\ No newline at end of file
+end