diff --git a/src/Generator.Tests/QueryHelpers.cs b/src/Generator.Tests/QueryHelpers.cs
index 268ec622..f8b6e0fc 100644
--- a/src/Generator.Tests/QueryHelpers.cs
+++ b/src/Generator.Tests/QueryHelpers.cs
@@ -5,6 +5,11 @@ namespace Generator.Tests
 {
     public static class LibraryQueryExtensions
     {
+        public static Namespace Namespace(this TranslationUnit unit, string name)
+        {
+            return unit.FindNamespace(name);
+        }
+
         public static Class Class(this Library library, string name)
         {
             return library.FindClass(name).ToList().First();