1 files changed, 4 insertions, 3 deletions

diff --git a/src/sp_disabled_functions.h b/src/sp_disabled_functions.h
index 258e2e6..ae1500a 100644
--- a/src/sp_disabled_functions.h
+++ b/src/sp_disabled_functions.h
@@ -9,9 +9,10 @@ bool should_disable(zend_execute_data *, const char *, const zend_string *,
                     const char *, const sp_list_node *, const zend_string *);
 bool should_disable_ht(zend_execute_data *, const char *, const zend_string *,
                        const char *, const sp_list_node *, const HashTable *);
-bool should_drop_on_ret_ht(zval *, const char *, const sp_list_node *config,
-                           const HashTable *);
-bool should_drop_on_ret(zval *, const sp_list_node *config, const char *);
+bool should_drop_on_ret_ht(const zval *, const char *, const sp_list_node *config,
+                           const HashTable *, zend_execute_data *);
+bool should_drop_on_ret(const zval *, const sp_list_node *config, const char *,
+                        zend_execute_data *);
 char *get_complete_function_path(zend_execute_data const *const);
 
 #endif /* __SP_DISABLE_FUNCTIONS_H */