Minor fix on tests, remove extra argument