Fix ArityException on data driven test