Fixed testcase with passing a function as the argument to a function