diff --git a/client/__tests__/e2e/e2eAnnotations.test.js b/client/__tests__/e2e/e2eAnnotations.test.js
index 14354033..1c4cee6e 100644
--- a/client/__tests__/e2e/e2eAnnotations.test.js
+++ b/client/__tests__/e2e/e2eAnnotations.test.js
@@ -129,6 +129,34 @@ describe.each([
     await colorByGeneset(meanExpressionBrushGenesetName);
     await assertColorLegendLabel(meanExpressionBrushGenesetName);
   });
+  test("diffexp", async () => {
+    if (config.withSubset) return;
+
+    await setup(config);
+
+    // set the two cell sets to b cells vs nk cells
+    await expandCategory(`louvain`);
+    await clickOn(`louvain:category-select`);
+    await clickOn(`categorical-value-select-louvain-B cells`);
+    await clickOn(`cellset-button-1`);
+    await clickOn(`categorical-value-select-louvain-B cells`);
+    await clickOn(`categorical-value-select-louvain-NK cells`);
+    await clickOn(`cellset-button-2`);
+
+    // check our assumption that we begin with 11
+    const _sets = await getAllByClass("geneset-expand");
+    expect(_sets).toHaveLength(11);
+
+    // run diffexp
+    await clickOn(`diffexp-button`);
+
+    // save
+    await waitByClass("autosave-complete");
+
+    // grab all the gene sets, check if the two new ones are there
+    const sets = await getAllByClass("geneset-expand");
+    expect(sets).toHaveLength(13);
+  });
   test("create a new geneset", async () => {
     if (config.withSubset) return;
 
diff --git a/client/src/components/geneExpression/index.js b/client/src/components/geneExpression/index.js
index 0670e751..cb4b5476 100644
--- a/client/src/components/geneExpression/index.js
+++ b/client/src/components/geneExpression/index.js
@@ -69,7 +69,7 @@ class GeneExpression extends React.Component {
             <H4
               role="menuitem"
               tabIndex="0"
-              data-testclass="category-expand"
+              data-testclass="geneset-heading-expand"
               onKeyPress={this.handleExpandGeneSets}
               style={{
                 cursor: "pointer",
diff --git a/client/src/components/geneExpression/quickGene.js b/client/src/components/geneExpression/quickGene.js
index 219345f1..aa414a98 100644
--- a/client/src/components/geneExpression/quickGene.js
+++ b/client/src/components/geneExpression/quickGene.js
@@ -122,7 +122,7 @@ function QuickGene() {
       <H4
         role="menuitem"
         tabIndex="0"
-        data-testclass="category-expand"
+        data-testclass="quickgene-heading-expand"
         onKeyPress={handleExpand}
         style={{
           cursor: "pointer",