diff --git a/atomsci/ddm/test/integrative/ad_index/test_ad_index.py b/atomsci/ddm/test/integrative/ad_index/test_ad_index.py
index 443700b3..7774f31b 100644
--- a/atomsci/ddm/test/integrative/ad_index/test_ad_index.py
+++ b/atomsci/ddm/test/integrative/ad_index/test_ad_index.py
@@ -35,7 +35,10 @@ def test():
     hp_params["python_path"] = python_path
     
     for feat in ['ecfp','mordred_filtered','rdkit_raw','graphconv']:
-        if feat in ['ecfp','graphconv']:
+        if feat == 'ecfp':
+            hp_params['featurizer']=feat
+        elif feat =='graphconv':
+            hp_params['model_type']='NN'
             hp_params['featurizer']=feat
         else:
             hp_params['featurizer']='computed_descriptors'