Fix test function signature.