diff --git a/container.go b/container.go
index cebe8273e..35bdfd781 100644
--- a/container.go
+++ b/container.go
@@ -96,7 +96,7 @@ type Container interface {
 	//
 	// errors:
 	// Systemerror - System error.
-	Set() error
+	Set(config configs.Config) error
 
 	// Start a process inside the container. Returns error if process fails to
 	// start. You can track process lifecycle with passed Process structure.
diff --git a/container_linux.go b/container_linux.go
index bf729760e..c44c8dacc 100644
--- a/container_linux.go
+++ b/container_linux.go
@@ -78,9 +78,10 @@ func (c *linuxContainer) Stats() (*Stats, error) {
 	return stats, nil
 }
 
-func (c *linuxContainer) Set() error {
+func (c *linuxContainer) Set(config configs.Config) error {
 	c.m.Lock()
 	defer c.m.Unlock()
+	c.config = &config
 	return c.cgroupManager.Set(c.config)
 }