diff --git a/backup.go b/backup.go
index b059250..8c49d22 100644
--- a/backup.go
+++ b/backup.go
@@ -15,7 +15,6 @@ import (
 
 	"github.com/docker/docker/api/types"
 	"github.com/docker/docker/api/types/container"
-	"github.com/docker/go-connections/nat"
 	"github.com/kennygrant/sanitize"
 	"github.com/spf13/cobra"
 )
@@ -23,9 +22,9 @@ import (
 // Backup is used to gather all of a container's metadata, so we can encode it
 // as JSON and store it
 type Backup struct {
-	Config  *container.Config
-	PortMap nat.PortMap
-	Mounts  []types.MountPoint
+	Config     *container.Config
+	Mounts     []types.MountPoint
+	HostConfig *container.HostConfig
 }
 
 var (
@@ -156,9 +155,9 @@ func backup(ID string) error {
 
 	paths = []string{}
 	backup := Backup{
-		PortMap: conf.HostConfig.PortBindings,
-		Config:  conf.Config,
-		Mounts:  conf.Mounts,
+		Config:     conf.Config,
+		HostConfig: conf.HostConfig,
+		Mounts:     conf.Mounts,
 	}
 
 	filename := sanitize.Path(fmt.Sprintf("%s-%s", conf.Config.Image, ID))
diff --git a/restore.go b/restore.go
index b32aece..027a3e9 100644
--- a/restore.go
+++ b/restore.go
@@ -10,7 +10,6 @@ import (
 	"strings"
 
 	"github.com/docker/docker/api/types"
-	"github.com/docker/docker/api/types/container"
 	"github.com/spf13/cobra"
 )
 
@@ -173,9 +172,7 @@ func createContainer(backup Backup) (string, error) {
 	}
 	// io.Copy(os.Stdout, reader)
 
-	resp, err := cli.ContainerCreate(ctx, backup.Config, &container.HostConfig{
-		PortBindings: backup.PortMap,
-	}, nil, "")
+	resp, err := cli.ContainerCreate(ctx, backup.Config, backup.HostConfig, nil, "")
 	if err != nil {
 		return "", err
 	}