diff --git a/test/unit/cmds/extension_cmds/create.test.js b/test/unit/cmds/extension_cmds/create.test.js
index 648c727a2..a39975a3f 100644
--- a/test/unit/cmds/extension_cmds/create.test.js
+++ b/test/unit/cmds/extension_cmds/create.test.js
@@ -1,6 +1,10 @@
 import test from 'ava'
 import { stub } from 'sinon'
 
+import {
+  emptyContext,
+  setContext
+} from '../../../../lib/context'
 import { successEmoji } from '../../../../lib/utils/emojis'
 
 import {
@@ -38,6 +42,12 @@ test.before(() => {
   }
   const createManagementClientStub = stub().returns(fakeClient)
 
+  emptyContext()
+  setContext({
+    cmaToken: 'mockedToken',
+    activeSpaceId: 'someSpaceId'
+  })
+
   createRewireAPI.__Rewire__('createManagementClient', createManagementClientStub)
   createRewireAPI.__Rewire__('success', successStub)
   logRewireAPI.__Rewire__('log', logStub)
diff --git a/test/unit/cmds/extension_cmds/delete.test.js b/test/unit/cmds/extension_cmds/delete.test.js
index c904146e9..cbb5c1956 100644
--- a/test/unit/cmds/extension_cmds/delete.test.js
+++ b/test/unit/cmds/extension_cmds/delete.test.js
@@ -1,6 +1,10 @@
 import test from 'ava'
 import { stub } from 'sinon'
 
+import {
+  emptyContext,
+  setContext
+} from '../../../../lib/context'
 import { successEmoji } from '../../../../lib/utils/emojis'
 
 import {
@@ -25,6 +29,12 @@ test.before(() => {
   }
   const createManagementClientStub = stub().returns(fakeClient)
 
+  emptyContext()
+  setContext({
+    cmaToken: 'mockedToken',
+    activeSpaceId: 'someSpaceId'
+  })
+
   deleteRewireAPI.__Rewire__('createManagementClient', createManagementClientStub)
   deleteRewireAPI.__Rewire__('success', successStub)
 })
diff --git a/test/unit/cmds/extension_cmds/get.test.js b/test/unit/cmds/extension_cmds/get.test.js
index 68bbd39e3..d54b79203 100644
--- a/test/unit/cmds/extension_cmds/get.test.js
+++ b/test/unit/cmds/extension_cmds/get.test.js
@@ -5,7 +5,10 @@ import {
   handler,
   __RewireAPI__ as getRewireAPI
 } from '../../../../lib/cmds/extension_cmds/get'
-
+import {
+  emptyContext,
+  setContext
+} from '../../../../lib/context'
 import {
   __RewireAPI__ as logRewireAPI
 } from '../../../../lib/cmds/extension_cmds/utils/log-as-table'
@@ -30,6 +33,12 @@ test.before(() => {
   }
   const createManagementClientStub = stub().returns(fakeClient)
 
+  emptyContext()
+  setContext({
+    cmaToken: 'mockedToken',
+    activeSpaceId: 'someSpaceId'
+  })
+
   getRewireAPI.__Rewire__('createManagementClient', createManagementClientStub)
   logRewireAPI.__Rewire__('log', logStub)
 })
diff --git a/test/unit/cmds/extension_cmds/list.test.js b/test/unit/cmds/extension_cmds/list.test.js
index 50934ccb7..97c381d2c 100644
--- a/test/unit/cmds/extension_cmds/list.test.js
+++ b/test/unit/cmds/extension_cmds/list.test.js
@@ -5,6 +5,10 @@ import {
   handler,
   __RewireAPI__ as rewireAPI
 } from '../../../../lib/cmds/extension_cmds/list'
+import {
+  emptyContext,
+  setContext
+} from '../../../../lib/context'
 
 const logStub = stub()
 const getSpaceStub = stub()
@@ -39,6 +43,12 @@ test.before(() => {
 
   const createManagementClientStub = stub().returns(fakeClient)
 
+  emptyContext()
+  setContext({
+    cmaToken: 'mockedToken',
+    activeSpaceId: 'someSpaceId'
+  })
+
   rewireAPI.__Rewire__('createManagementClient', createManagementClientStub)
   rewireAPI.__Rewire__('log', logStub)
 })
@@ -48,7 +58,7 @@ test.after.always(() => {
   rewireAPI.__ResetDependency__('log')
 })
 
-test('Lists extensions', async (t) => {
+test.serial('Lists extensions', async (t) => {
   await handler({spaceId: 'space'})
 
   const outputValues = [ 'Widget', '123', 'Widget 2', '456' ]
@@ -58,7 +68,7 @@ test('Lists extensions', async (t) => {
   })
 })
 
-test('Displays message if list is empty', async (t) => {
+test.serial('Displays message if list is empty', async (t) => {
   await handler({spaceId: 'empty-space'})
 
   t.true(logStub.calledWith('No extensions found'))
diff --git a/test/unit/cmds/extension_cmds/update.test.js b/test/unit/cmds/extension_cmds/update.test.js
index b482721e5..86882089c 100644
--- a/test/unit/cmds/extension_cmds/update.test.js
+++ b/test/unit/cmds/extension_cmds/update.test.js
@@ -7,7 +7,10 @@ import {
   updateExtension,
   __RewireAPI__ as rewireAPI
 } from '../../../../lib/cmds/extension_cmds/update'
-
+import {
+  emptyContext,
+  setContext
+} from '../../../../lib/context'
 import { ValidationError } from '../../../../lib/utils/error'
 
 const updateStub = stub()
@@ -32,6 +35,12 @@ test.before(() => {
   }
   const createManagementClientStub = stub().returns(fakeClient)
 
+  emptyContext()
+  setContext({
+    cmaToken: 'mockedToken',
+    activeSpaceId: 'someSpaceId'
+  })
+
   rewireAPI.__Rewire__('createManagementClient', createManagementClientStub)
   rewireAPI.__Rewire__('logExtension', logStub)
   rewireAPI.__Rewire__('success', successStub)