Support module def in module function

yck1509 · Nov 20, 2015 · d4b35d2 · d4b35d2
1 parent f9bdba9
commit d4b35d2
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Confuser.Core/Project/Patterns/ModuleFunction.cs b/Confuser.Core/Project/Patterns/ModuleFunction.cs
@@ -20,9 +20,11 @@ public override int ArgumentCount {
 
 		/// <inheritdoc />
 		public override object Evaluate(IDnlibDef definition) {
-			if (!(definition is IOwnerModule))
+			if (!(definition is IOwnerModule) && !(definition is IModule))
 				return false;
 			object name = Arguments[0].Evaluate(definition);
+			if (definition is IModule)
+				return ((IModule)definition).Name == name.ToString();
 			return ((IOwnerModule)definition).Module.Name == name.ToString();
 		}
 	}