Convert function to arrow function in UT