diff --git a/momento/cache.w b/momento/cache.w
index d64bffd9..c9841305 100644
--- a/momento/cache.w
+++ b/momento/cache.w
@@ -43,7 +43,7 @@ pub class Cache {
     let defaultTtl = props.defaultTtl ?? 60s;
     if target.startsWith("tf") {
       this.inner = new Cache_tf({ token, name, defaultTtl });
-    } if target == "sim" {
+    } elif target == "sim" {
       this.inner = new Cache_sim({ token, name, defaultTtl });
     } else {
       throw "unsupported target: " + target;
@@ -155,6 +155,9 @@ class MomentoProvider {
       name: "momento",
       source: "momentohq/momento",
       version: "0.1.0",
+      attributes: {
+        api_key: util.env("MOMENTO_API_KEY"),
+      }
     ) as singletonKey in root;
   }
 }
diff --git a/momento/package.json b/momento/package.json
index e76622f5..a358591a 100644
--- a/momento/package.json
+++ b/momento/package.json
@@ -1,7 +1,7 @@
 {
   "name": "@winglibs/momento",
   "description": "momento library for Wing",
-  "version": "0.0.1",
+  "version": "0.0.2",
   "repository": {
     "type": "git",
     "url": "https://github.com/winglang/winglibs.git",
@@ -24,4 +24,4 @@
     "@gomomento/sdk": "^1.93.0",
     "@winglibs/tf": "^0.0.5"
   }
-}
+}
\ No newline at end of file