diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
index 3061862d..5bd038f0 100644
--- a/.github/workflows/run_tests.yml
+++ b/.github/workflows/run_tests.yml
@@ -30,7 +30,7 @@ jobs:
       - name: install pymc and poetry
         run: |
           mamba info
-          mamba install -c conda-forge pymc=5.14 poetry
+          mamba install -c conda-forge pymc poetry
 
       - name: install hssm
         run: |
diff --git a/src/hssm/hssm.py b/src/hssm/hssm.py
index 6eb14ce4..97c520cf 100644
--- a/src/hssm/hssm.py
+++ b/src/hssm/hssm.py
@@ -502,7 +502,7 @@ def sample(
         # in the GLM as far as bambi is concerned)
         if self._inference_obj is not None:
             if self._parent not in self._inference_obj.posterior.data_vars.keys():
-                self.sample_posterior_predictive(self._inference_obj, kind="mean")
+                self.model.predict(self._inference_obj, kind="mean", inplace=True)
                 # rename 'rt,response_mean' to 'v' so in the traces everything
                 # looks the way it should
                 self._inference_obj.rename_vars(