Fix test function call