Fix tests so they provide correct function args