diff --git a/batchglm/unit_test/test_acc_constrained_vglm_all.py b/batchglm/unit_test/test_acc_constrained_vglm_all.py
index 2222fe5d..1723a325 100644
--- a/batchglm/unit_test/test_acc_constrained_vglm_all.py
+++ b/batchglm/unit_test/test_acc_constrained_vglm_all.py
@@ -24,8 +24,11 @@ def simulate(self):
         constraints[1, 1] = 1
         constraints[2, 2] = 1
         constraints[3, 2] = -1
+        new_coef_names = ['Intercept', 'condition[T.1]', 'batch[1]', 'batch[2]']
         self.sim1.input_data.design_loc = dmat
         self.sim1.input_data.design_scale = dmat
+        self.sim1.input_data._design_loc_names = new_coef_names
+        self.sim1.input_data._design_scale_names = new_coef_names
         self.sim1.input_data.constraints_loc = constraints
         self.sim1.input_data.constraints_scale = constraints
 
diff --git a/batchglm/unit_test/test_acc_glm_all.py b/batchglm/unit_test/test_acc_glm_all.py
index a27ef339..66680cbb 100644
--- a/batchglm/unit_test/test_acc_glm_all.py
+++ b/batchglm/unit_test/test_acc_glm_all.py
@@ -50,6 +50,8 @@ def __init__(
                 data=scipy.sparse.csr_matrix(simulator.input_data.x),
                 design_loc=simulator.input_data.design_loc,
                 design_scale=simulator.input_data.design_scale,
+                design_loc_names=simulator.input_data.design_loc_names,
+                design_scale_names=simulator.input_data.design_scale_names,
                 constraints_loc=simulator.input_data.constraints_loc,
                 constraints_scale=simulator.input_data.constraints_scale,
                 size_factors=simulator.input_data.size_factors
@@ -59,6 +61,8 @@ def __init__(
                 data=simulator.input_data.x,
                 design_loc=simulator.input_data.design_loc,
                 design_scale=simulator.input_data.design_scale,
+                design_loc_names=simulator.input_data.design_loc_names,
+                design_scale_names=simulator.input_data.design_scale_names,
                 constraints_loc=simulator.input_data.constraints_loc,
                 constraints_scale=simulator.input_data.constraints_scale,
                 size_factors=simulator.input_data.size_factors