1 files changed, 6 insertions, 4 deletions
diff --git a/src/compiler/IntermediateRep.cc b/src/compiler/IntermediateRep.cc
index 6cb6580283..1567fccf1a 100644
--- a/src/compiler/IntermediateRep.cc
+++ b/src/compiler/IntermediateRep.cc
@@ -22,12 +22,14 @@ namespace art {
 /* Allocate a new basic block */
 BasicBlock* oatNewBB(CompilationUnit* cUnit, BBType blockType, int blockId)
 {
-    BasicBlock* bb = (BasicBlock* )oatNew(sizeof(BasicBlock), true, kAllocBB);
+    BasicBlock* bb = (BasicBlock* )oatNew(cUnit, sizeof(BasicBlock), true,
+                                          kAllocBB);
     bb->blockType = blockType;
     bb->id = blockId;
-    bb->predecessors = (GrowableList*) oatNew(sizeof(GrowableList), false,
-                                              kAllocPredecessors);
-    oatInitGrowableList(bb->predecessors, (blockType == kExitBlock) ? 2048 : 2,
+    bb->predecessors = (GrowableList*) oatNew(cUnit, sizeof(GrowableList),
+                                              false, kAllocPredecessors);
+    oatInitGrowableList(cUnit, bb->predecessors,
+                        (blockType == kExitBlock) ? 2048 : 2,
                         kListPredecessors);
     return bb;
 }