diff --git a/src/hotspot/share/gc/shared/modRefBarrierSet.hpp b/src/hotspot/share/gc/shared/modRefBarrierSet.hpp
index 7195faf..a94eb15 100644
--- a/src/hotspot/share/gc/shared/modRefBarrierSet.hpp
+++ b/src/hotspot/share/gc/shared/modRefBarrierSet.hpp
@@ -32,9 +32,9 @@ class Klass;
 
 class ModRefBarrierSet: public BarrierSet {
 protected:
-  ModRefBarrierSet(BarrierSetCodeGen* code_gen,
+  ModRefBarrierSet(BarrierSetAssembler* barrier_set_assembler,
                    const BarrierSet::FakeRtti& fake_rtti)
-    : BarrierSet(code_gen,
+    : BarrierSet(barrier_set_assembler,
                  fake_rtti.add_tag(BarrierSet::ModRef)) { }
   ~ModRefBarrierSet() { }