diff --git a/ci_test/unit_tests/test_unit_layer_channelwise_softmax_distconv.py b/ci_test/unit_tests/test_unit_layer_channelwise_softmax_distconv.py
index 60f8369ac0d..0db96d05d58 100644
--- a/ci_test/unit_tests/test_unit_layer_channelwise_softmax_distconv.py
+++ b/ci_test/unit_tests/test_unit_layer_channelwise_softmax_distconv.py
@@ -184,5 +184,5 @@ def construct_data_reader(lbann):
 # ==============================================
 
 # Create test functions that can interact with PyTest
-for _test_func in tools.create_tests(setup_experiment, __file__):
+for _test_func in tools.create_tests(setup_experiment, __file__, environment=tools.get_distconv_environment()):
     globals()[_test_func.__name__] = _test_func