diff --git a/runkit_methods.c b/runkit_methods.c
index 3acb4d3..2f74837 100644
--- a/runkit_methods.c
+++ b/runkit_methods.c
@@ -62,7 +62,7 @@ zend_class_entry *_php_runkit_locate_scope(zend_class_entry *ce, zend_function *
 #else
 /* {{{ _php_runkit_get_method_prototype
 	Locates the prototype method */
-zend_function* _php_runkit_get_method_prototype(zend_class_entry *ce, char* func, int func_len) {
+zend_function* _php_runkit_get_method_prototype(zend_class_entry *ce, char* func, int func_len TSRMLS_DC) {
 	zend_class_entry *pce = ce;
 	zend_function *proto = NULL;
     char *funcLower;
@@ -413,7 +413,7 @@ static void php_runkit_method_add_or_update(INTERNAL_FUNCTION_PARAMETERS, int ad
 	func.common.function_name = estrndup(methodname, methodname_len);
 #ifdef ZEND_ENGINE_2
 	func.common.scope = ce;
-	func.common.prototype = _php_runkit_get_method_prototype(ce, methodname, methodname_len);
+	func.common.prototype = _php_runkit_get_method_prototype(ce, methodname, methodname_len TSRMLS_CC);
 
 	if (flags & ZEND_ACC_PRIVATE) {
 		func.common.fn_flags &= ~ZEND_ACC_PPP_MASK;
@@ -496,7 +496,7 @@ static int php_runkit_method_copy(char *dclass, int dclass_len, char *dfunc, int
 
 #ifdef ZEND_ENGINE_2
 	dfe.common.scope = dce;
-	dfe.common.prototype = _php_runkit_get_method_prototype(dce, dfunc, dfunc_len);
+	dfe.common.prototype = _php_runkit_get_method_prototype(dce, dfunc, dfunc_len TSRMLS_CC);
 #endif
 
 	if (zend_hash_add(&dce->function_table, dfuncLower, dfunc_len + 1, &dfe, sizeof(zend_function), &dfeInHashTable) == FAILURE) {