diff --git a/Packages/ohif-hotkeys/client/classes/HotkeysManager.js b/Packages/ohif-hotkeys/client/classes/HotkeysManager.js
index 3383c1235..ffeb4ffca 100644
--- a/Packages/ohif-hotkeys/client/classes/HotkeysManager.js
+++ b/Packages/ohif-hotkeys/client/classes/HotkeysManager.js
@@ -33,7 +33,7 @@ export class HotkeysManager {
         const storageKey = `hotkeysDefinitions.${contextName}`;
         return new Promise((resolve, reject) => {
             if (this.storeFunction) {
-                this.storeFunction(contextName, definitions).then(resolve).catch(reject);
+                this.storeFunction.call(this, storageKey, definitions).then(resolve).catch(reject);
             } else if (Meteor.userId()) {
                 OHIF.user.setData(storageKey, definitions).then(resolve).catch(reject);
             } else {
diff --git a/Packages/ohif-viewerbase/client/lib/WLPresets.js b/Packages/ohif-viewerbase/client/lib/WLPresets.js
index 01fefce3a..05702655f 100644
--- a/Packages/ohif-viewerbase/client/lib/WLPresets.js
+++ b/Packages/ohif-viewerbase/client/lib/WLPresets.js
@@ -50,6 +50,14 @@ class WindowLevelPresetsManager {
         this.changeObserver = new Tracker.Dependency();
     }
 
+    setRetrieveFunction(retrieveFunction) {
+        this.retrieveFunction = retrieveFunction;
+    }
+
+    setStoreFunction(storeFunction) {
+        this.storeFunction = storeFunction;
+    }
+
     updateElementWLPresetData(element) {
         const wlPresetData = cornerstone.getElementData(element, 'wlPreset');
         const enabledElement = cornerstone.getEnabledElement(element);
@@ -123,7 +131,7 @@ class WindowLevelPresetsManager {
     store(wlPresets) {
         return new Promise((resolve, reject) => {
             if (this.storeFunction) {
-                this.storeFunction.call(this, wlPresets).then(resolve).catch(reject);
+                this.storeFunction.call(this, WL_STORAGE_KEY, wlPresets).then(resolve).catch(reject);
             } else if (Meteor.userId()) {
                 OHIF.user.setData(WL_STORAGE_KEY, wlPresets).then(resolve).catch(reject);
             } else {