diff --git a/test/swagger.test.js b/test/swagger.test.js
index d689d09f1..18a6dc10c 100644
--- a/test/swagger.test.js
+++ b/test/swagger.test.js
@@ -89,7 +89,10 @@ describe('swagger command', () => {
       .reply(200, [{ version: '1.0.1' }])
       .post('/api/v1/version')
       .basicAuth({ user: key })
-      .reply(200, { version: '1.0.1' });
+      .reply(200, { version: '1.0.1' })
+      .post('/api/v1/api-specification', body => body.match('form-data; name="spec"'))
+      .basicAuth({ user: key })
+      .reply(201, { id: 1 });
 
     return swagger(['./test/fixtures/swagger.json'], { key, version }).then(() => mock.done());
   });
diff --git a/test/versions.test.js b/test/versions.test.js
index 2783d3776..9ab0779c4 100644
--- a/test/versions.test.js
+++ b/test/versions.test.js
@@ -178,7 +178,7 @@ describe('Versions CLI Commands', () => {
       const mockRequest = nock(config.host)
         .delete(`/api/v1/version/${version}`)
         .basicAuth({ user: key })
-        .reply(201);
+        .reply(200);
 
       await deleteVersion([], { key, version });
       mockRequest.done();