diff --git a/configs/learning_gem5/part3/msi_caches.py b/configs/learning_gem5/part3/msi_caches.py
index 5aa7abed24..691b5fdcbf 100644
--- a/configs/learning_gem5/part3/msi_caches.py
+++ b/configs/learning_gem5/part3/msi_caches.py
@@ -111,7 +111,8 @@ class MyCacheSystem(RubySystem):
             self.sequencers[i].connectCpuPorts(cpu)
 
 
-class L1Cache(L1Cache_Controller):
+class L1Cache(MSI_L1Cache_Controller):
+
     _version = 0
 
     @classmethod
@@ -176,7 +177,8 @@ class L1Cache(L1Cache_Controller):
         self.responseFromDirOrSibling.in_port = ruby_system.network.out_port
 
 
-class DirController(Directory_Controller):
+class DirController(MSI_Directory_Controller):
+
     _version = 0
 
     @classmethod
diff --git a/configs/learning_gem5/part3/ruby_caches_MI_example.py b/configs/learning_gem5/part3/ruby_caches_MI_example.py
index 583041a674..6eaec7276c 100644
--- a/configs/learning_gem5/part3/ruby_caches_MI_example.py
+++ b/configs/learning_gem5/part3/ruby_caches_MI_example.py
@@ -109,7 +109,8 @@ class MyCacheSystem(RubySystem):
             self.sequencers[i].connectCpuPorts(cpu)
 
 
-class L1Cache(L1Cache_Controller):
+class L1Cache(MI_example_L1Cache_Controller):
+
     _version = 0
 
     @classmethod
@@ -165,7 +166,8 @@ class L1Cache(L1Cache_Controller):
         self.responseToCache.in_port = ruby_system.network.out_port
 
 
-class DirController(Directory_Controller):
+class DirController(MI_example_Directory_Controller):
+
     _version = 0
 
     @classmethod