diff --git a/src/__mocks__/model-universe/index.ts b/src/__mocks__/model-universe/index.ts
index 5f0facb9f..ec8be30a3 100644
--- a/src/__mocks__/model-universe/index.ts
+++ b/src/__mocks__/model-universe/index.ts
@@ -12,7 +12,11 @@ export class MockModel implements ModelPluginInterface {
   configure(): Promise<ModelPluginInterface> {
     return Promise.resolve(this);
   }
-  execute(): Promise<any[]> {
+  execute(inputs: any): Promise<any[]> {
+    if (inputs[0].carbon) {
+      return Promise.resolve(inputs);
+    }
+
     return Promise.resolve([{data: 'mock-data'}]);
   }
 }