diff --git a/test/units/TestModules.py b/test/units/TestModules.py
index 83c2b7c3986..aef2e83ed62 100644
--- a/test/units/TestModules.py
+++ b/test/units/TestModules.py
@@ -16,7 +16,7 @@ class TestModules(unittest.TestCase):
             for (dirpath, dirnames, filenames) in os.walk(path):
                 for filename in filenames:
                     (path, ext) = os.path.splitext(filename)
-                    if ext != ".ps1":
+                    if ext == ".py":
                         module_list.append(os.path.join(dirpath, filename))
         return module_list