Fix args issue in sample extension