Fix changes of function names in test