diff --git a/tests/test_optProb.py b/tests/test_optProb.py
index fe061638..fb7cb625 100644
--- a/tests/test_optProb.py
+++ b/tests/test_optProb.py
@@ -10,9 +10,9 @@
 
 try:
     # External modules
-    import mpi4py  # noqa:F401
+    from mpi4py import MPI
 
-    comm = mpi4py.MPI.COMM_WORLD
+    comm = MPI.COMM_WORLD
 except ImportError:
     comm = None