diff --git a/src/ports/postgres/modules/recursive_partitioning/random_forest.py_in b/src/ports/postgres/modules/recursive_partitioning/random_forest.py_in
index 930d916..1226591 100644
--- a/src/ports/postgres/modules/recursive_partitioning/random_forest.py_in
+++ b/src/ports/postgres/modules/recursive_partitioning/random_forest.py_in
@@ -332,10 +332,7 @@
                 cat_features, ordered_cat_features, con_features, boolean_cats = \
                     _classify_features(all_cols_types, features)
 
-                filter_null = _get_filter_str(schema_madlib, cat_features,
-                                              con_features, boolean_cats,
-                                              dependent_variable, grouping_cols,
-                                              max_n_surr)
+                filter_null = _get_filter_str(dependent_variable, grouping_cols)
                 # the total number of records
                 n_all_rows = plpy.execute("SELECT count(*) FROM {0}".
                                           format(training_table_name))[0]['count']