diff --git a/app/scripts/metamask-controller.actions.test.js b/app/scripts/metamask-controller.actions.test.js
index 268ab1f67ca5..b017c4d642fe 100644
--- a/app/scripts/metamask-controller.actions.test.js
+++ b/app/scripts/metamask-controller.actions.test.js
@@ -253,7 +253,7 @@ describe('MetaMaskController', function () {
           Promise.resolve({ supportsInterface: supportsInterfaceStub }),
         );
       sinon
-        .stub(metamaskController.tokensController, 'getNetworkClientById')
+        .stub(metamaskController.controllerMessenger, 'call')
         .callsFake(() => ({
           configuration: {
             chainId: '0xa',
@@ -267,8 +267,11 @@ describe('MetaMaskController', function () {
         networkClientId: 'networkClientId1',
       });
       assert.strictEqual(
-        metamaskController.tokensController.getNetworkClientById.getCall(0)
-          .args[0],
+        metamaskController.controllerMessenger.call.getCall(0).args[0],
+        'NetworkController:getNetworkClientById',
+      );
+      assert.strictEqual(
+        metamaskController.controllerMessenger.call.getCall(0).args[1],
         'networkClientId1',
       );
     });