1 files changed, 2 insertions, 4 deletions
diff --git a/src/main/scala/firrtl/passes/memlib/ResolveMemoryReference.scala b/src/main/scala/firrtl/passes/memlib/ResolveMemoryReference.scala
index 783c179f..108f6fd4 100644
--- a/src/main/scala/firrtl/passes/memlib/ResolveMemoryReference.scala
+++ b/src/main/scala/firrtl/passes/memlib/ResolveMemoryReference.scala
@@ -29,10 +29,8 @@ object ResolveMemoryReference extends Pass {
     case s => s map updateMemStmts(uniqueMems)
   }
 
-  def updateMemMods(m: DefModule) = {
+  def run(c: Circuit) = {
     val uniqueMems = new AnnotatedMemories
-    (m map updateMemStmts(uniqueMems))
+    c copy (modules = c.modules map (_ map updateMemStmts(uniqueMems)))
   }
-
-  def run(c: Circuit) = c copy (modules = c.modules map updateMemMods)
 }