diff --git a/builder/amazon/chroot/step_check_root_device.go b/builder/amazon/chroot/step_check_root_device.go
index d4d202f25..da18599aa 100644
--- a/builder/amazon/chroot/step_check_root_device.go
+++ b/builder/amazon/chroot/step_check_root_device.go
@@ -12,7 +12,7 @@ import (
 type StepCheckRootDevice struct{}
 
 func (s *StepCheckRootDevice) Run(state multistep.StateBag) multistep.StepAction {
-	image := state.Get("ec2").(*ec2.Image)
+	image := state.Get("source_image").(*ec2.Image)
 	ui := state.Get("ui").(packer.Ui)
 
 	ui.Say("Checking the root device on source AMI...")