Fix args bug