diff --git a/stoptime.rb b/stoptime.rb
index ee67dc3..5722e55 100644
--- a/stoptime.rb
+++ b/stoptime.rb
@@ -1059,6 +1059,7 @@ module StopTime::Controllers
         @customer.save
         if @customer.invalid?
           @errors = @customer.errors
+          @button = "update"
           return render :customer_form
         end
       end