diff --git a/pyproject.toml b/pyproject.toml
index 7644b51..a825c44 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -40,7 +40,7 @@ httpx = "^0.24.1"
 
 
 [tool.poe.tasks]
-install-pyciemss = "pip install --no-cache-dir git+https://github.com/ciemss/pyciemss.git@f5c5bec731441377b845a3b702f01927113bf462 --use-pep517"
+install-pyciemss = "pip install --no-cache-dir git+https://github.com/ciemss/pyciemss.git@e224fd5737e96a7a64ed9b0512ca4084eb7a4c87 --use-pep517"
 
 
 [tool.pytest.ini_options]
diff --git a/service/models/operations/optimize.py b/service/models/operations/optimize.py
index 9ce386a..7edc414 100644
--- a/service/models/operations/optimize.py
+++ b/service/models/operations/optimize.py
@@ -28,7 +28,7 @@ class QOI(BaseModel):
     contexts: List[str] = []
 
     def gen_call(self):
-        contexts = [context + "_state" for context in self.contexts]
+        contexts = self.contexts
         qoi_map = {
             QOIMethod.day_average: lambda samples: obs_nday_average_qoi(
                 samples, contexts, 1
diff --git a/tests/examples/optimize/input/request.json b/tests/examples/optimize/input/request.json
index 9050584..b6c142d 100644
--- a/tests/examples/optimize/input/request.json
+++ b/tests/examples/optimize/input/request.json
@@ -15,7 +15,7 @@
 		"end": 90
 	},
 	"qoi": {
-		"contexts": ["Infected"],
+		"contexts": ["Infected_state"],
 		"method": "day_average"
 	},
 	"risk_bound": 10.0,