diff --git a/addon/validated-buffer.js b/addon/validated-buffer.js
index <HASH>..<HASH> 100644
--- a/addon/validated-buffer.js
+++ b/addon/validated-buffer.js
@@ -1,11 +1,7 @@
 import EmberValidations from 'ember-validations';
 import BufferedProxy from 'ember-buffered-proxy/proxy';
 
-export default function validatedBuffer(object, classBody, container) {
-  if (container) {
-    object.set('container', container);
-  }
-
+export default function validatedBuffer(object, classBody) {
   var Buffer = BufferedProxy.extend(EmberValidations.Mixin, classBody);
 
   return Buffer.create({
require the object to have a container property