diff --git a/cmd/restore.go b/cmd/restore.go
index 270e090..af65b83 100644
--- a/cmd/restore.go
+++ b/cmd/restore.go
@@ -72,5 +72,16 @@ func restoreRunE(_ *cobra.Command, args []string) error {
 	spaceWorker := space_worker.NewSpaceWorker(logger.LogVerbose(flagVerbose), workDir)
 	spaceWorker.AnsibleBin = flagAnsibleBin
 	spaceWorker.AnsibleVerbose = flagAnsibleVerbose
-	return spaceWorker.Restore(flagTargetHost, flagRestoreTimestamp, flagRestoreInventory, args...)
+
+	var restoreInventory string
+	if flagRestoreInventory != "" {
+		restoreInventory = flagRestoreInventory
+	} else {
+		restoreInventory, err = spaceWorker.GetDefaultHost()
+		if err != nil {
+			return err
+		}
+	}
+
+	return spaceWorker.Restore(flagTargetHost, flagRestoreTimestamp, restoreInventory, args...)
 }