diff --git a/src/query_graph.ts b/src/query_graph.ts
index feb19f7d..5b0f2b68 100644
--- a/src/query_graph.ts
+++ b/src/query_graph.ts
@@ -373,6 +373,8 @@ export default class QueryGraph {
         ) {
           nodes[qNodeID].categories.push('biolink:Protein');
         }
+        // Ensure categories are rolled into expandedCategories
+        nodes[qNodeID].expandCategories()
       }
     }
     this.logs.push(