diff --git a/test/test_learn.py b/test/test_learn.py
index 0c8dee4c..74a26cd3 100644
--- a/test/test_learn.py
+++ b/test/test_learn.py
@@ -99,7 +99,6 @@ def test_learn_3d_reg():
             chain2='D',
             mapfly=False,
             use_rotation=2,
-            grid_shape=(30, 30, 30),
             select_feature={
                 'AtomicDensities_ind': 'all',
                 'Feature_ind': [ 'coulomb', 'vdwaals', 'charge', 'PSSM_*']},