Fixed small args bug