diff --git a/pkg/extend/extend.go b/pkg/extend/extend.go
index a5d489156..0e1a8e501 100644
--- a/pkg/extend/extend.go
+++ b/pkg/extend/extend.go
@@ -65,7 +65,7 @@ func extend(d, fsType string) error {
 
 	data, err := exec.Command("sfdisk", "-J", d).Output()
 	if err != nil {
-		log.Fatalf("Unable to get drive data for %s from sfdisk: %v", d, err)
+		return fmt.Errorf("Unable to get drive data for %s from sfdisk: %v", d, err)
 	}
 
 	f := Fdisk{}