diff --git a/jest/mock.js b/jest/mock.js
index ace189e..19e732f 100644
--- a/jest/mock.js
+++ b/jest/mock.js
@@ -2,10 +2,15 @@ const { NativeModules, DeviceEventEmitter } = require('react-native')
 
 if (!NativeModules.RNLlama) {
   NativeModules.RNLlama = {
+    setContextLimit: jest.fn(),
+
+    modelInfo: jest.fn(async () => ({})),
+
     initContext: jest.fn(() =>
       Promise.resolve({
         gpu: false,
         reasonNoGPU: 'Test',
+        model: {},
       }),
     ),