diff --git a/containerd/containerd.go b/containerd/containerd.go
index 5ea3a66..9dc7e81 100644
--- a/containerd/containerd.go
+++ b/containerd/containerd.go
@@ -136,6 +136,7 @@ func (d *Driver) createContainer(containerConfig *ContainerConfig, config *TaskC
 	var opts []oci.SpecOpts
 
 	if config.Entrypoint != nil {
+		opts = append(opts, oci.WithImageConfig(containerConfig.Image))
 		// WithProcessArgs replaces the args on the generated spec.
 		opts = append(opts, oci.WithProcessArgs(args...))
 	} else {