diff --git a/src/ram/Aggregate.h b/src/ram/Aggregate.h
index 5d1d5291597..06b663e7849 100644
--- a/src/ram/Aggregate.h
+++ b/src/ram/Aggregate.h
@@ -72,6 +72,7 @@ class Aggregate : public RelationOperation, public AbstractAggregate {
         for (auto &elem : orderBy) {
             elem->expr = map(std::move(elem->expr));
         }
+        function->apply(map);
     }
 
     static bool classof(const Node* n) {
diff --git a/src/ram/IndexAggregate.h b/src/ram/IndexAggregate.h
index 32a8251db0a..b715975ce85 100644
--- a/src/ram/IndexAggregate.h
+++ b/src/ram/IndexAggregate.h
@@ -74,6 +74,7 @@ class IndexAggregate : public IndexOperation, public AbstractAggregate {
         for (auto &elem : orderBy) {
             elem->expr = map(std::move(elem->expr));
         }
+        function->apply(map);
     }
 
     static bool classof(const Node* n) {