diff --git a/DDTest/python/test_import_ddg4.py b/DDTest/python/test_import_ddg4.py
index 122a1e4f21cf348cfede5d70058e5b5c1439e5cc..da77c50f6ead06663840b8dc82d0e0a5d237ba91 100644
--- a/DDTest/python/test_import_ddg4.py
+++ b/DDTest/python/test_import_ddg4.py
@@ -5,7 +5,11 @@ This file adds a test for the DDG4 python module.
 """
 
 from __future__ import absolute_import
-from test_import import test_module, parametrize
+import pytest
+from test_import import test_module
+
+
+parametrize = pytest.mark.parametrize
 
 
 moduleNames = [
@@ -15,4 +19,4 @@ moduleNames = [
 
 @parametrize('moduleName', moduleNames)
 def test_module_ddg4(moduleName):
-    test_module(moduleName)
+    test_module('DDG4')