diff --git a/test/aws.storage.test.js b/test/aws.storage.test.js
index 8dbb468a..61fe5657 100644
--- a/test/aws.storage.test.js
+++ b/test/aws.storage.test.js
@@ -112,11 +112,10 @@ describe('Testing Set using aws', function() {
     sinon.stub(crypto, 'randomBytes').returns(buf);
     s3Stub.upload.callsArgWith(1, null, {});
     return storage
-      .set('123', {}, 'Filename.moz', 'url.com')
-      .then(reply => {
-        assert.equal(reply.uuid, buf.toString('hex'));
-        assert.equal(reply.url, 'url.com');
-        assert.notEqual(reply.uuid, null);
+      .set('123', {}, 'Filename.moz', {})
+      .then(deleteKey => {
+        assert.equal(deleteKey, buf.toString('hex'));
+        assert.notEqual(deleteKey, null);
         assert(expire.calledOnce);
         assert(expire.calledWith('123', 86400000));
       })
diff --git a/test/local.storage.test.js b/test/local.storage.test.js
index 659710a6..d965d8cc 100644
--- a/test/local.storage.test.js
+++ b/test/local.storage.test.js
@@ -122,10 +122,9 @@ describe('Testing Set to local filesystem', function() {
     fsStub.createWriteStream.returns({ on: stub });
 
     return storage
-      .set('test', { pipe: sinon.stub() }, 'Filename.moz', 'moz.la')
-      .then(reply => {
-        assert(reply.uuid);
-        assert.equal(reply.url, 'moz.la');
+      .set('test', { pipe: sinon.stub() }, 'Filename.moz', {})
+      .then(deleteKey => {
+        assert(deleteKey);
       })
       .catch(err => assert.fail());
   });