From 6686b623475caf81be9c06a38dca8e4db6e53955 Mon Sep 17 00:00:00 2001
From: Nevins Bartolomeo <nevins.bartolomeo@aocsolutions.com>
Date: Fri, 3 Apr 2015 20:52:54 -0400
Subject: [PATCH] AWS availability zone could be incorrect when using spot
 instances with no AZ specified

---
 builder/amazon/common/step_run_source_instance.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/builder/amazon/common/step_run_source_instance.go b/builder/amazon/common/step_run_source_instance.go
index 50cedf6ea..f114e156b 100644
--- a/builder/amazon/common/step_run_source_instance.go
+++ b/builder/amazon/common/step_run_source_instance.go
@@ -75,6 +75,7 @@ func (s *StepRunSourceInstance) Run(state multistep.StateBag) multistep.StepActi
 	}
 
 	spotPrice := s.SpotPrice
+	availabilityZone := s.AvailabilityZone
 	if spotPrice == "auto" {
 		ui.Message(fmt.Sprintf(
 			"Finding spot price for %s %s...",
@@ -96,6 +97,7 @@ func (s *StepRunSourceInstance) Run(state multistep.StateBag) multistep.StepActi
 		}
 
 		var price float64
+
 		for _, history := range resp.History {
 			log.Printf("[INFO] Candidate spot price: %s", history.SpotPrice)
 			current, err := strconv.ParseFloat(history.SpotPrice, 64)
@@ -105,6 +107,9 @@ func (s *StepRunSourceInstance) Run(state multistep.StateBag) multistep.StepActi
 			}
 			if price == 0 || current < price {
 				price = current
+				if s.AvailabilityZone == "" {
+					availabilityZone = history.AvailabilityZone
+				}
 			}
 		}
 		if price == 0 {
@@ -158,7 +163,7 @@ func (s *StepRunSourceInstance) Run(state multistep.StateBag) multistep.StepActi
 			SubnetId:                 s.SubnetId,
 			AssociatePublicIpAddress: s.AssociatePublicIpAddress,
 			BlockDevices:             s.BlockDevices.BuildLaunchDevices(),
-			AvailZone:                s.AvailabilityZone,
+			AvailZone:                availabilityZone,
 		}
 		runSpotResp, err := ec2conn.RequestSpotInstances(runOpts)
 		if err != nil {