diff --git a/diffxpy/testing/tests.py b/diffxpy/testing/tests.py
index c378494..6c208ce 100644
--- a/diffxpy/testing/tests.py
+++ b/diffxpy/testing/tests.py
@@ -694,7 +694,10 @@ def wald(
                 as_numeric=as_numeric
             )
         else:
-            coef_loc_names = dmat_loc.columns.tolist()
+            if isinstance(dmat_loc, patsy.design_info.DesignMatrix):
+                coef_loc_names = dmat_loc.design_info.column_names
+            else:
+                coef_loc_names = dmat_loc.columns.tolist()
         if not np.all([x in coef_loc_names for x in coef_to_test]):
             raise ValueError(
                 "the requested test coefficients %s were found in model coefficients %s" %
diff --git a/docs/tutorials.rst b/docs/tutorials.rst
index cdcbfd0..872ef1d 100644
--- a/docs/tutorials.rst
+++ b/docs/tutorials.rst
@@ -36,9 +36,7 @@ Diffxpy allows you to define a data set partition and to conduct test on each ge
 Multiple tests per gene
 ~~~~~~~~~~~~~~~~~~~~~~~
 
-How to perform `pairwise <https://nbviewer.jupyter.org/github/theislab/diffxpy_tutorials/tree/master/diffxpy_tutorials/test/multiple_tests_per_gene.ipynb>`__ tests.
-
-How to perform group tests versus all other groups (versus `rest <https://nbviewer.jupyter.org/github/theislab/diffxpy_tutorials/tree/master/diffxpy_tutorials/test/multiversus_rest.ipynb>`__).
+How to perform `pairwise tests, group versus rest tests and tests within each parition  <https://nbviewer.jupyter.org/github/theislab/diffxpy_tutorials/tree/master/diffxpy_tutorials/test/multiple_tests_per_gene.ipynb>`__ tests.
 
 
 Gene set enrichment: enrich